input = torch.randn(20, 2, 11, 11, 32)输出形式

input = torch.randn(20, 2, 11, 11, 32)
m = torch.nn.AdaptiveAvgPool3d((1,1, 32))
x=m(input)
print(x.shape)

结果：

也就是不用管批次和通道数