評價此頁

torch.Tensor.stride#

Tensor.stride(dim) tuple or int#

返回 self 張量的步幅。

步幅是在指定維度 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