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)

结果:

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

你可能感兴趣的:(深度学习,python,机器学习)