快捷方式

torch.Tensor.stride

Tensor.stride(dim) tuple or int

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获取问题解答

查看资源