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