torch.Tensor.map_

Tensor.map_(tensor, callable)

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

callable 应具有以下签名：

def callable(a, b) -> number