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