torch.Tensor.set_

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

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

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

参数

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

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

• size (torch.Size, 可选) – 指定的目标大小，默认与源大小相同。

• stride (元组, 可选) – 指定的步长。默认为 C 顺序的连续步长。