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 连续的步长。