torch.Tensor.map_

Tensor.map_(tensor, callable)

将 callable 应用于 self 张量和给定 tensor 中的每个元素，并将结果存储在 self 张量中。self 张量和给定 tensor 必须是可广播的 (broadcastable)。

可调用对象 (callable) 应该具有以下签名：

def callable(a, b) -> number