torch.Tensor.map_

Tensor.map_(tensor, callable)

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

callable 应具有以下签名

def callable(a, b) -> number