mindspore的nn.Conv3D 怎么指定纬度padding

问题描述：

x = Tensor(np.ones([16, 3, 10, 32, 32]), mindspore.float32)

conv3d = nn.Conv3d(in_channels=3, out_channels=32, kernel_size=(4, 3, 3), pad_mode='pad', padding=(2,2,3))

output = conv3d(x)

print(output.shape)

请问，我想对后三纬度进行指定填充，该怎么做呢？ 上面代码会报错

解答：

padding的数据类型是一个整数或六个整数的元组。

x = Tensor(np.ones([13, 3, 10, 32, 32]), mindspore.float32)

# 方法一：padding=一个整数

conv3d = nn.Conv3d(in_channels=3, out_channels=32, kernel_size=(4, 3, 3), pad_mode='pad', padding=3)



# 方法二：padding=六个整数的元组

conv3d = nn.Conv3d(in_channels=3, out_channels=32, kernel_size=(4, 3, 3), pad_mode='pad', padding=(3,3,2,2,2,2))



output = conv3d(x)

print(output.shape)