torch.Tensor.map_

Tensor.map_(tensor, callable)

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

该 callable 应该具有签名

def callable(a, b) -> number