[ torch 参数更多 ]torch.max_pool1d¶
torch.max_pool1d¶
torch.max_pool1d(input, kernel_size, stride=[], padding=[0], dilation=[1], ceil_mode=False)
paddle.nn.functional.max_pool1d¶
paddle.nn.functional.max_pool1d(x, kernel_size, stride=None, padding=0, return_mask=False, ceil_mode=False, 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_mask | 是否返回最大值的索引,Paddle 保持默认即可。 |