torch.Tensor.set_¶
- Tensor.set_(source=None, storage_offset=0, size=None, stride=None) Tensor ¶
设置底层存储、大小和步幅。如果
source
是一个张量,则self
张量将共享相同的存储,并具有与source
相同的大小和步幅。一个张量中元素的更改将反映在另一个张量中。如果
source
是一个Storage
,则该方法设置底层存储、偏移量、大小和步幅。- 参数
source (Tensor 或 Storage) – 要使用的张量或存储
storage_offset (int, 可选) – 存储中的偏移量
size (torch.Size, 可选) – 期望的大小。默认为源的大小。
stride (tuple, 可选) – 期望的步幅。默认为 C 连续步幅。