torch.Tensor.map_

Tensor.map_(tensor, callable)

Применяет callable к каждому элементу тензора self и заданному tensor и сохраняет результаты в тензоре self. Тензор self и заданный tensor должны быть совместимыми для трансляции.

Функция callable должна иметь сигнатуру:

def callable(a, b) -> number