megengine.functional.nn.conv_transpose3d

conv_transpose3d(inp, weight, bias=None, stride=1, padding=0, output_padding=0, dilation=1, groups=1)[source]

3D transposed convolution operation. Only support the case that groups = 1 and conv_mode = “cross_correlation”.

Refer to ConvTranspose3d for more information.

Parameters

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

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

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

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

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

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

  • dilation (Union[int, Tuple[int, int, int]]) – dilation of the 3D 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, depth, height, width). Default: 1

Return type

Tensor

Returns

output tensor.

previous

megengine.functional.nn.conv_transpose2d

next

megengine.functional.nn.deformable_conv2d