如何查看pytorch里面tensor的形状信息？

import torch

torch = torch.randn(2,3,4)

print(torch.shape())

报错：TypeError: 'torch.Size' object is not callable

改用 print(torch.size())
    print(torch.size(0))
    print(torch.size(1))
               ···
               ···
               ···
    print(torch.size(n))
只要你的张量有n维，你就可以打印某一维的size有多大，也就是，该维度有多少个数据

参考资料

TypeError: 'torch.Size' object is not callable

TypeError: 'torch.Size' object is not callable_wang xiang的博客-CSDN博客