mx.nd.Convolution.v1¶
Description¶
This operator is DEPRECATED. Apply convolution to input then add a bias.
Arguments¶
Argument |
Description |
|---|---|
|
NDArray-or-Symbol. Input data to the ConvolutionV1Op. |
|
NDArray-or-Symbol. Weight matrix. |
|
NDArray-or-Symbol. Bias parameter. |
|
Shape(tuple), required. convolution kernel size: (h, w) or (d, h, w) |
|
Shape(tuple), optional, default=[]. convolution stride: (h, w) or (d, h, w) |
|
Shape(tuple), optional, default=[]. convolution dilate: (h, w) or (d, h, w) |
|
Shape(tuple), optional, default=[]. pad for convolution: (h, w) or (d, h, w) |
|
int (non-negative), required. convolution filter(channel) number |
|
int (non-negative), optional, default=1. Number of group partitions. Equivalent to slicing input into num_group partitions, apply convolution on each, then concatenate the results |
|
long (non-negative), optional, default=1024. Maximum temporary workspace allowed for convolution (MB).This parameter determines the effective batch size of the convolution kernel, which may be smaller than the given batch size. Also, the workspace will be automatically enlarged to make sure that we can run the kernel with batch_size=1 |
|
boolean, optional, default=0. Whether to disable bias parameter. |
|
{None, ‘fastest’, ‘limited_workspace’, ‘off’},optional, default=’None’. Whether to pick convolution algo by running performance test. Leads to higher startup time but may give faster speed. Options are: ‘off’: no tuning ‘limited_workspace’: run test and pick the fastest algorithm that doesn’t exceed workspace limit. ‘fastest’: pick the fastest algorithm and ignore workspace limit. If set to None (default), behavior is determined by environment variable MXNET_CUDNN_AUTOTUNE_DEFAULT: 0 for off, 1 for limited workspace (default), 2 for fastest. |
|
boolean, optional, default=0. Turn off cudnn for this layer. |
|
{None, ‘NCDHW’, ‘NCHW’, ‘NDHWC’, ‘NHWC’},optional, default=’None’. Set layout for input, output and weight. Empty for default layout: NCHW for 2d and NCDHW for 3d. |
Value¶
out The result mx.ndarray