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