評價此頁

torch.Tensor.copy_#

Tensor.copy_(src, non_blocking=False) Tensor#

src 的元素複製到 self 張量中,並返回 self

src 張量必須與 self 張量 相容廣播。它可以是不同的資料型別或位於不同的裝置上。

引數
  • src (Tensor) – 要從中複製的源張量

  • non_blocking (bool, optional) – 如果為 True 並且此複製是在 CPU 和 GPU 之間進行的,則複製可能相對於主機非同步進行。對於其他情況,此引數沒有影響。預設為 False