• 1. 正常情况下,判断一个Tensor的维度是通过 tensor.shape 进行判断,例如:
import torch

x = torch.randint(low=-10,high=10,size=(3,6))
print(x)
print(x.shape)

 

由此可以判断是2维(第1维有3个元素,第2维有6个元素)。

  • 2. 但有时候,需要在代码中判断某一个Tensor到底是1维还是2维还是3维等,该如何判断呢?可以通过 len(tensor.shape) 判断,例如:
import torch

x = torch.randint(low=-10,high=10,size=(3,6))
print(x)
print(x.shape)
print(len(x.shape))

 

则会输出维度是2。

 

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐