本文主要是介绍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)
结果:
也就是不用管批次和通道数
这篇关于input = torch.randn(20, 2, 11, 11, 32)输出形式的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!