mindspore.Tensor.stride

查看源文件
mindspore.Tensor.stride(dim=None)[源代码]

在指定维度 dim中 从一个元素跳到下一个元素所必需的步长。当没有参数传入时,返回所有维度的步长的列表。

参数:
  • dim (int) - 指定的维度。

返回:

int,返回在指定维度下,从一个元素调到下一个元素所必需的步长。

异常:
  • TypeError - dim 不是int。

样例:

>>> import mindspore as ms
>>> x = ms.Tensor([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]], dtype=ms.float32)
>>> x.stride()
[5, 1]