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

2023-12-14 18:06:02
input = torch.randn(20, 2, 11, 11, 32)
m = torch.nn.AdaptiveAvgPool3d((1,1, 32))
x=m(input)
print(x.shape)

结果：

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

文章来源:https://blog.csdn.net/Kongxiangyunltj/article/details/134907126
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。