如何在 PyTorch 中获取张量的维数（也称为阶数或度数）？

Question

如果我有一个 PyTorch 张量，例如

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

如何获得该张量的维数？在这种情况下，它将是 4。

Answer 1

从 PyTorch 中，张量的维数存储在张量的

.ndim

属性中，或者您可以调用

.dim()

函数。

如果你有

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

您可以通过

获得维度数

t.ndim 
# 4

或

t.dim()
# 4