Python PyTorch Tensor.map_用法及代码示例

本文简要介绍python语言中 torch.Tensor.map_ 的用法。

用法:

Tensor.map_(tensor, callable)

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

callable 应该有签名：

def callable(a, b) -> number