torch.Tensor.apply_

Tensor.apply_(callable) → Tensor

Применяет функцию callable к каждому элементу тензора, заменяя каждый элемент значением, возвращаемым callable.

Примечание

Эта функция работает только с тензорами CPU и не должна использоваться в участках кода, требующих высокой производительности.