nn.Conv2d在PTorch和MindSpore中padding参数的映射问题

在PTtorch中padding=(3,0)表示只填充高、宽两个维度:

nn.Conv2d在PTorch和MindSpore中padding参数的映射问题_第1张图片

而在MindSpore中padding=(3,0),则会报错,如何设置才能达到和在Ptorch中的设置效果呢?

下面是MindSpor中的padding参数的解释,这里的上下左右填充和Ptorch中高宽的解释是不是一样呢?

根据官方文档, mindspore.nn.Conv2d 中的 padding 必须是int或包含4个整数的tuple,如果 padding 是一个有4个整数的tuple,那么上、下、左、右的填充分别等于 padding[0] 、 padding[1] 、 padding[2] 和 padding[3] 。

因此,在你的用例中,应该传入 padding=(3, 3, 0, 0)。同时,你还需要传入 pad_mode="pad"。

你可能感兴趣的:(深度学习,人工智能,python,ubuntu)