Python PyTorch Tensor.stride用法及代碼示例

本文簡要介紹python語言中 torch.Tensor.stride 的用法。

用法:

Tensor.stride(dim) → tuple or int

參數：

dim(int,可選的) -需要步幅的所需尺寸

返回self 張量的步幅。

步幅是在指定維度 dim 中從一個元素到下一個元素所需的跳躍。如果沒有傳入參數，則返回所有步幅的元組。否則，返回一個整數值作為特定維度 dim 中的步幅。

例子：

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