torch.Tensor.stride

Tensor.stride(dim) → 元组或整型

返回 self 张量的步长。

步长是从一个元素跳到下一个元素所需的跳跃量 dim 。当不传递任何参数时，返回所有步长的元组。否则，返回特定维度的步长整数值 dim 。

参数:

dim（int，可选）- 需要步长的期望维度

示例：

>>> x = torch.tensor([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]])
>>> x.stride()
(5, 1)
>>> x.stride(0)
5
>>> x.stride(-1)
1