torch.Tensor.map_

Tensor.map_(张量, 可调用对象)

self张量和给定的tensor中的每个元素应用callable函数,并将结果存储在self张量中。self张量和给定的tensor必须是广播兼容的。

callable 应该具有如下签名:

def callable(a, b) -> number
本页目录