megengine.functional.nn.conv_transpose3d

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

三维转置卷积运算。只支持 group=1 且 conv_mode="cross_correlation"。

更多信息参见 ConvTranspose3d。

参数

• inp (Tensor) – 卷积运算的特征图。

• weight (Tensor) – 卷积核。shape 通常是 (in_channels, out_channels, height, width).

• bias (Optional[Tensor]) – 添加到卷积结果中的偏置量(如果给定该值)

• stride (Union[int, Tuple[int, int, int]]) – 三维卷积运算中的步长。 默认： 1

• padding (Union[int, Tuple[int, int, int]]) – 输入数据空域维度每侧的填充（padding）大小。仅支持填充0值。默认：0

• output_padding (Union[int, Tuple[int, int, int]]) – padding 到输出的 size。默认：0

• dilation (Union[int, Tuple[int, int, int]]) – 三维卷积运算的扩张值(dilation)。 默认： 1

• groups (int) – 在进行“分组卷积”时，对输入输出通道的分组数量。当 groups 不是 1 时，in_channels 和 out_channels 必须可被 groups 整除，卷积核权重的形状将会是 (groups, in_channels // groups, out_channels // groups, depth, height, width) 。默认值：1

返回类型

Tensor

返回

输出张量。

上一页

megengine.functional.nn.conv_transpose2d

下一页

megengine.functional.nn.deformable_conv2d