Pad 差异对比

torch.nn.functional.pad

torch.nn.functional.pad(
   input,
   pad,
   mode='constant',
   value=None
)

更多请查看 torch.nn.functional.pad.

megengine.module.Pad

megengine.module.Pad(
   pad_width,
   mode='constant',
   constant_val=0.0

)

更多请查看 megengine.module.Pad.

参数差异

pad

PyTorch pad 参数表示一个 one-hot 向量的长度，MegEngine pad_width 与之对应，表示一个元组。元组中的每个元素都是两个元素的元组.