ConvTranspose3d 差异对比

torch.nn.ConvTranspose3d

torch.nn.ConvTranspose3d(
   in_channels,
   out_channels,
   kernel_size,
   stride=1,
   padding=0,
   output_padding=0,
   groups=1,
   bias=True,
   dilation=1,
   padding_mode='zeros',
   device=None,
   dtype=None
)

更多请查看 torch.nn.ConvTranspose3d.

megengine.module.ConvTranspose3d

megengine.module.ConvTranspose3d(
   in_channels,
   out_channels,
   kernel_size,
   stride=1,
   padding=0,
   output_padding=0,
   dilation=1,
   groups=1,
   bias=True,
   conv_mode='cross_correlation'
   compute_mode='default'
   ** kwargs
)

更多请查看 megengine.module.ConvTranspose3d.

使用差异

padding

PyTorch padding 可以是单个数字或元组，MegEngine padding 仅支持数值填充 0.

compute_mode 参数

MegEngine 中包含 compute_mode 参数，PyTorch 中无此参数，该参数用于指定计算模式，当设置 “default” 时, 不会对中间结果的精度有特殊要求。当设置 “float32” 时, “float32” 将被用作中间结果的累加器, 但是只有当输入和输出的 dtype 是 float16 时有效。

conv_mode 参数

MegEngine 中包含 conv_mode 参数，PyTorch 中无此参数，该参数用于指定卷积模式，默认值为 “cross_correlation” 时, 表示使用交叉相关计算模式。在这种模式下，输入张量被视为滤波器的目标图像，而滤波器在输入上进行滑动以生成输出张量。