快捷方式

torch.Tensor.map_

Tensor.map_(tensor, callable)

callable 應用於 self 張量中的每個元素,以及給定的 tensor,並將結果儲存到 self 張量中。self 張量和給定的 tensor 必須是 可廣播的

callable 應具有以下簽名:

def callable(a, b) -> number

文件

取得 PyTorch 的完整開發者文件

查看文件

教學

取得為初學者和進階開發者設計的深入教學課程

查看教學

資源

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

查看資源