torch.Tensor.stride

Tensor.stride(dim) → tuple 或 int

返回 self 张量的步长。

步长是移动到指定维度 dim 中下一个元素所需的步进量。未传入参数时，返回所有维度的步长组成的 tuple。否则，返回指定维度 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