之前的pytorch代码使用的是二维池化操作对一个最后一维为1的张量进行操作，比如self.pool = nn.AvgPool2d(kernel_size=(2, 1), stride=(2, 1))，相当于是一个一维池化，但是在jittor的文档中没有找到一维的池化函数，jittor库中的二维池化函数不支持张量维度等于kernelsize的操作。
之前的pytorch代码使用的是二维池化操作对一个最后一维为1的张量进行操作，比如self.pool = nn.AvgPool2d(kernel_size=(2, 1), stride=(2, 1))，相当于是一个一维池化，但是在jittor的文档中没有找到一维的池化函数，jittor库中的二维池化函数不支持张量维度等于kernelsize的操作。