Pytorch nn.MaxPool1d; nn.functional.max_pool1d

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

参数:

kernel_size(int or tuple) - max pooling的窗口大小
stride(int or tuple, optional) - max pooling的窗口移动的步长。默认值是kernel_size
padding(int or tuple, optional) - 输入的每一条边补充0的层数
dilation(int or tuple, optional) – 一个控制窗口中元素步幅的参数
return_indices - 如果等于True,会返回输出最大值的序号,对于上采样操作会有帮助
ceil_mode - 如果等于True,计算输出信号大小的时候,会使用向上取整,代替默认的向下取整的操作

shape:
输入: (N,C,L_in)
输出: (N,C,L_out)
L o u t = f l o o r ( ( L i n + 2 p a d d i n g − d i l a t i o n ( k e r n e l s i z e − 1 ) − 1 ) / s t r i d e + 1 L_{out}=floor((L_{in} + 2padding - dilation(kernelsize - 1) - 1)/stride + 1 Lout=floor((Lin+2paddingdilation(kernelsize1)1)/stride+1

你可能感兴趣的:(Pytorch)