（beta）)torch_npu.contrib.function.npu_fast_condition_index_put

接口原型

npu_fast_condition_index_put(x, condition, value):

功能描述

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

参数说明

x (torch.Tensor) - normal Tensor。
condition (torch.BoolTensor) - 判断条件。
value (Int, Float) - bboxes步长。

输出说明

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