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博客