torch.Tensor.apply_

Tensor.apply_(callable) Tensor

对张量中的每个元素应用函数 callable,并将每个元素替换成该函数返回的值。

注意

此函数仅支持CPU张量,不应在需要高性能的代码段中使用。

本页目录