算子功能:将输入的self张量视为一维数组,把index的值当作索引,从self中取值,输出shape与index一致的Tensor。
计算公式:其中下标i表示从0遍历到index的元素个数-1。
每个算子分为,必须先调用“aclnnTakeGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnTake”接口执行计算。
[object Object]
[object Object]
参数说明
[object Object]- [object Object]Atlas 训练系列产品[object Object]:self和out的数据类型不支持BFLOAT16。
返回值
第一段接口完成入参校验,出现以下场景时报错:
[object Object]
- 确定性计算:
- aclnnTake默认确定性实现。
[object Object]