Paddle 对于形状相同的输入,nn.MaxPool2D的输出形状与PyTorch有差异

jchrr9hc  于 5个月前  发布在  其他
关注(0)|答案(1)|浏览(44)

bug描述 Describe the Bug

input = paddle.randn([1, 30, 15, 20])
pool = paddle.nn.MaxPool2D(kernel_size=[3, 3], stride=[5, 6], padding=[0, 0], ceil_mode=True)
output = pool(input)
print(output.shape)

输出

[1, 30, 4, 4]

对于同样的代码,pytorch输出的形状不同

input = torch.randn([1, 30, 15, 20])
pool = torch.nn.MaxPool2d(kernel_size=[3, 3], stride=[5, 6], padding=[0, 0], dilation=[1, 1], ceil_mode=True)
output = pool(input)
print(output.shape)

输出

torch.Size([1, 30, 3, 4])

其他补充信息 Additional Supplementary Information

paddle版本2.6.1
torch版本2.1.1

lsmd5eda

lsmd5eda1#

torch的测试里面你设置了dilation,Paddle的API并没有这个参数

相关问题