[torch 参数更多]torch.nn.functional.max_pool3d

torch.nn.functional.max_pool3d

torch.nn.functional.max_pool3d(input, kernel_size, stride=None, padding=0, dilation=1, ceil_mode=False, return_indices=False)

paddle.nn.functional.max_pool3d

paddle.nn.functional.max_pool3d(x, kernel_size, stride=None, padding=0, ceil_mode=False, return_mask=False, data_format="NCDHW", name=None)

Pytorch 相比 Paddle 支持更多其他参数,具体如下:

参数映射

PyTorch PaddlePaddle 备注
input x 输入的 Tensor,仅参数名不一致。
kernel_size kernel_size 池化核的尺寸大小。
stride stride 池化操作步长。
padding padding 池化补零的方式。
dilation - 带有滑动窗口元素间的 stride,Paddle 无此参数,暂无转写方式。
ceil_mode ceil_mode 是否用 ceil 函数计算输出的 height 和 width。
return_indices return_mask 是否返回最大值的索引,仅参数名不一致。
- data_format 输入和输出的数据格式,PyTorch 无此参数,Paddle 保持默认即可。