def npu_fast_condition_index_put()

API接口

def npu_fast_condition_index_put(x, condition, value):

功能描述

使用NPU亲和写法替换bool型index_put函数中的原生写法。

参数说明

输出说明

torch.Tensor - 框转换deltas。

示例

调用方式示例：
from torch_npu.contrib.function import npu_fast_condition_index_put
x1_opt = npu_fast_condition_index_put(x, condition, value)
使用示例：
    >>> x = torch.randn(128, 8192)
    >>> condition = x < 0.5
    >>> value = 0.
    >>> x1 = copy.deepcopy(x)[condition] = value
    >>> x1_opt = npu_fast_condition_index_put(x, condition, value)
父主题： torch_npu.contrib