torch.Tensor.map_¶
- Tensor.map_(tensor, callable)¶
Applies
callable
for each element inself
tensor and the giventensor
and stores the results inself
tensor.self
tensor and the giventensor
must be broadcastable.The
callable
should have the signature:def callable(a, b) -> number