如何获取PyTorch张量的维度数（即阶数）？

嘿，这事儿其实超简单～有好几种轻松的方法能拿到PyTorch张量的维度数，就拿你给出的这个张量例子来说：

import torch
t = torch.rand((2,3,4,5))

我给你挨个介绍常用的几种方式：

• 使用dim()方法：这是最直接的方式，直接调用张量的dim()方法就能返回维度数，代码如下：

  print(t.dim())  # 输出结果：4
  

• 使用ndim属性：这是PyTorch较新版本里支持的属性，用法更简洁，直接访问就行：

  print(t.ndim)   # 输出结果：4
  

• 通过size()或shape属性的长度获取：张量的size()和shape都会返回一个包含各维度大小的元组，元组的长度就是张量的维度数，比如：

  print(len(t.size()))  # 输出结果：4
  print(len(t.shape))   # 输出结果：4
  

选哪种都可以，前两种更直观，看你个人习惯啦～

备注：内容来源于stack exchange，提问作者Oskar