torch.Tensor.map_

Tensor.map_(tensor, callable)

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

callable 应有以下签名：

def callable(a, b) -> number