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]