operator[]

产品支持情况

产品	是否支持
Atlas 350 加速卡	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品AI Core	√
Atlas 推理系列产品Vector Core	√
Atlas 训练系列产品	√

功能说明

获取距原LocalTensor起始地址偏移量为offset的新LocalTensor，注意offset不能超过原有LocalTensor的size大小。

函数原型

__aicore__ inline LocalTensor operator[](const uint32_t offset) const

参数说明

表1 参数说明
参数名	输入/输出	描述
offset	输入	偏移量，单位为元素。

返回值说明

返回距原LocalTensor起始地址偏移量为offset的新LocalTensor。

约束说明

无

调用示例

// srcLen = 256, num = 100, M=50
// operator[]使用方法, inputLocal[16]为从起始地址开始偏移量为16的新tensor
AscendC::Add(outputLocal[16], inputLocal[16], inputLocal2[16], M);
// 示例结果如下：
// 输入数据(inputLocal): [100 100 100 ... 100]
// 输入数据(inputLocal2): [1 2 3 ... 66]
// 输出数据(outputLocal): [... 117 118 119 ... 166]

父主题： LocalTensor