torch.Tensor.set_

Tensor.set_(source=None, storage_offset=0, size=None, stride=None) → Tensor

設定底層儲存、大小和步長。如果 source 是一個 tensor，則 self tensor 將與 source 共享相同的儲存，並擁有相同的大小和步長。對其中一個 tensor 中元素的更改將反映在另一個 tensor 中。

如果 source 是一個 Storage，此方法將設定底層儲存、偏移量、大小和步長。

引數

• source (Tensor 或 Storage) – 要使用的 tensor 或 storage

• storage_offset (int, 可選) – 在儲存中的偏移量

• size (torch.Size, 可選) – 期望的大小。預設為 source 的大小。

• stride (tuple, 可選) – 期望的步長。預設為 C 連續的步長。