torch.Tensor.map_

Tensor.map_(tensor, callable)

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

The callable 应该具有以下签名：

def callable(a, b) -> number