快捷方式

torch.Tensor.set_

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

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

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

参数
  • source (TensorStorage) – 要使用的张量或存储

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

  • size (torch.Size, 可选) – 期望的大小。默认为源的大小。

  • stride (tuple, 可选) – 期望的步幅。默认为 C 连续步幅。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答疑问

查看资源