megengine.functional.nn.conv_transpose2d

conv_transpose2d(inp, weight, bias=None, stride=1, padding=0, output_padding=0, dilation=1, groups=1, conv_mode='cross_correlation', compute_mode='default')[source]

2D transposed convolution operation.

Refer to ConvTranspose2d for more information.

Parameters

  • inp (Tensor) – feature map of the convolution operation.

  • weight (Tensor) – convolution kernel. weight usually has shape (in_channels, out_channels, height, width).

  • bias (Optional[Tensor]) – bias added to the result of convolution (if given).

  • stride (Union[int, Tuple[int, int]]) – stride of the 2D convolution operation. Default: 1

  • padding (Union[int, Tuple[int, int]]) – size of the paddings added to the input on both sides of its spatial dimensions. Only zero-padding is supported. Default: 0

  • output_padding (Union[int, Tuple[int, int]]) – size of paddings appended to output. Default: 0

  • dilation (Union[int, Tuple[int, int]]) – dilation of the 2D convolution operation. Default: 1

  • groups (int) – number of groups into which the input and output channels are divided, so as to perform a grouped convolution. When groups is not 1, in_channels and out_channels must be divisible by groups, and the shape of weight should be (groups, in_channels // groups, out_channels // groups, height, width). Default: 1

  • conv_mode – supports “cross_correlation”. Default: “cross_correlation”

  • compute_mode – when set to “default”, no special requirements will be placed on the precision of intermediate results. When set to “float32”, “float32” would be used for accumulator and intermediate result, but only effective when input and output are of float16 dtype.

Return type

Tensor

Returns

output tensor.

previous

megengine.functional.nn.local_conv2d

next

megengine.functional.nn.conv_transpose3d