IndexFill

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×

功能说明

• 算子功能：沿输入x的给定轴dim，将indices指定位置的值使用value进行非原地填充替换。

• 示例： 输入x为:

    [[1, 2, 3],

     [4, 5, 6],

     [7, 8, 9]]

  若dim = 0，indices = [0, 2]，value = 0时，算子的计算结果为：

    [[0, 0, 0],

     [4, 5, 6],

     [0, 0, 0]]

  若dim = 1，indices = [0, 2]，value = 0时，算子的计算结果为：

    [[0, 2, 0],

     [0, 5, 0],

     [0, 8, 0]]

参数说明

参数名	输入/输出	描述	数据类型	数据格式
x	输入	功能示例中的x，即待被在指定位置的值用value非原地替换的张量。	FLOAT16、FLOAT、DOUBLE、INT8、UINT8、INT16、INT32、INT64、BOOL、BFLOAT16	ND
indices	输入	指定x在dim维度将要填充的下标。其中的元素值小于x对应dim的维度大小，支持负数索引。	INT32、INT64	ND
value	输入	指定填充的数据值。数据类型需与x一致。	与x一致	ND
dim	属性	指定了x将要填充的维度。dim的取值范围在[-x.dim(), x.dim())。	INT64	-
y	输出	填充后的输出张量，shape与输入x一致。	与输入x一致	ND

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_index_fill	通过aclnnIndexFill接口方式调用IndexFill算子。