评价此页

torch.Tensor.stride#

Tensor.stride(dim) tuple or int#

返回张量的跨度。

跨度是从指定维度 dim 中一个元素跳到下一个元素所需的步长。如果不传递参数,则返回所有跨度的元组。否则,将返回指定维度 dim 的整数跨度值。

参数

dim (int, optional) – 所需跨度的维度

示例

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