FilesLast commitLast update
22 days ago
4 months ago
22 days ago
1 month ago
24 days ago
24 days ago
22 days ago
24 days ago
1 month ago
README

Index

产品支持情况

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

功能说明

  • 算子功能:按 indices 指定的下标对输入 x 进行多维高级索引(advanced indexing),取出对应坐标的数据得到输出 y。各输入与功能的关系如下:
    • indexed_sizes:一维 INT64 张量,长度等于 x 的维度数,逐维标记 x 的每一维是否参与索引:取值为 1 表示该维由索引选择(消耗一个 indices 输入),取值大于 1 表示该维整体保留。
    • indexed_strides:一维 INT64 张量,长度等于 x 的维度数,给出按索引取数时在 x 中每一维上的步长。
    • indices:动态输入,个数等于 indexed_sizes 中取值为 1 的维度数量,作为对应被索引维度的下标;多个 indices 之间按广播规则对齐。
    • y:按上述规则从 x 中取出的数据,数据类型与 x 一致。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 输入数据。 FLOAT、FLOAT16、BFLOAT16、INT32、INT64、INT8、UINT8、BOOL、COMPLEX64 ND
indexed_sizes 输入 索引个数。 INT64 ND
indexed_strides 输入 索引步长。 INT64 ND
indices 输入 索引。 INT32、INT64 ND
y 输出 根据索引取出后的数据。 FLOAT、FLOAT16、BFLOAT16、INT32、INT64、INT8、UINT8、BOOL、COMPLEX64 ND
  • Atlas训练系列产品、Atlas推理系列产品:数据类型不支持BFLOAT16、COMPLEX64。
  • Atlas A2系列产品、Atlas A3系列产品:不支持COMPLEX64。

约束说明

  • Ascend 950PR&950DT系列产品:indices各tensor之间的shape需相同或满足广播关系。
  • 其他芯片:indices各tensor之间的shape需完全相同。

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_index 通过aclnnIndex接口方式调用Index算子。