torch.Tensor.dim_order

Tensor.dim_order() → tuple

返回一个整数元组，描述 self 的维度顺序或物理布局。

参数

无 –

维度顺序表示维度在内存中的布局方式，从最外层到最内层维度。

示例：

>>> torch.empty((2, 3, 5, 7)).dim_order()
(0, 1, 2, 3)
>>> torch.empty((2, 3, 5, 7), memory_format=torch.channels_last).dim_order()
(0, 2, 3, 1)

警告

dim_order 张量 API 处于实验阶段，可能会发生变化。