torch.Tensor.set_

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

设置底层存储、大小和步幅。如果 source 是一个张量， self 张量将共享相同的存储，并具有与 source 相同的大小和 步幅。对一个张量中元素的更改将反映在另一个张量中。

如果 source 是一个 Storage，该方法会设置底层存储、偏移量、大小和步幅。

Parameters

• 源 (Tensor 或 Storage) – 要使用的张量或存储

• storage_offset (int, 可选) – 存储中的偏移量

• 大小 (torch.Size, 可选) – 所需的大小。默认为源的大小。

• 步幅 (元组, 可选) – 所需的步幅。默认为C连续步幅。