torch.Tensor.stride

Tensor.stride(dim) → tuple or int

返回 self 张量的步幅。

步幅是在指定维度 dim 中，从一个元素移动到下一个元素所需的跳跃量。当不传递任何参数时，返回一个包含所有步幅的元组。否则，将返回指定维度 dim 的步幅整数值。

参数

dim (int, optional) – 需要步幅的目标维度

示例

>>> 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