Maxpool1D
函数原型:
torch.nn.MaxPool1d(kernel_size: Union[T, Tuple[T, ...]],
stride: Optional[Union[T, Tuple[T, ...]]] = None,
padding: Union[T, Tuple[T, ...]] = 0,
dilation: Union[T, Tuple[T, ...]] = 1,
return_indices: bool = False,
ceil_mode: bool = False)
kernel
kernel 大小为 1 * kernel_sizestride
步长默认为 kernel_size (一个kernel里取一个最大值,然后移动到下一个kernel)padding
两边补充 -INF 的个数dilation:
不懂return_indices:
返回最大值的下标,一个kernel里返回一个ceil_mode:
是否向上取整,即,如果剩下的不足一个kernel,是舍弃还是padding到一个kernel的大小
Maxpool2D
基本同Maxpool1D,注意:
- kernel大小为 kernel_size * kernel_size(或者是自定义的元组)
- 元组(a,b)a为第一维上的操作,b为第二维上的操作
(这里的第一,第二是相对意义上的第一第二,不是真正的第一维或第二维)
版权声明:本文为weixin_44134372原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。