torch.Tensor.apply_

Tensor.apply_(callable) → Tensor

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

Примечание

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