torch.Tensor.dim_order

Tensor.dim_order() → 元组[源代码]

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

参数

无 –

Dim顺序表示维度在内存中的排列方式，从外到内的顺序。

示例:

>>> 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 是试验性的，可能随时发生变化。