快速鍵

torch.Tensor.copy_

Tensor.copy_(src, non_blocking=False) Tensor

將元素從 src 複製到 self 張量中,並傳回 self

src 張量必須與 self 張量 可廣播。它可以是不同的資料類型或駐留在不同的裝置上。

參數
  • src (Tensor) – 要複製的來源張量

  • non_blocking (bool) – 如果 True 且此複製操作發生在 CPU 和 GPU 之間,則複製可能相對於主機以非同步方式發生。 對於其他情況,此參數無效。

文件

取得 PyTorch 的完整開發人員文件

檢視文件

教學

取得適用於初學者和進階開發人員的深入教學

檢視教學

資源

尋找開發資源並取得問題解答

檢視資源