昇腾社区首页
中文
注册

operator()

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 200I/500 A2 推理产品

x

Atlas 推理系列产品AI Core

x

Atlas 推理系列产品Vector Core

x

Atlas 训练系列产品

x

Atlas 200/300/500 推理产品

x

功能说明

获取某个索引位置的元素的引用。

LocalTensor的operator()类似。

函数原型

1
__aicore__ inline __inout_pipe__(S) __gm__ PrimType& operator()(const uint64_t offset) const

参数说明

表1 参数说明

参数名

输入/输出

描述

offset

输入

GlobalTensor的下标索引。

返回值说明

返回指定索引位置的元素的PrimType类型引用。

PrimType定义如下:

1
2
// PrimT用于从T中提取基础数据类型:T传入基础数据类型,直接返回数据类型;T传入为TensorTrait类型时萃取TensorTrait中的LiteType基础数据类型
using PrimType = PrimT<T>;

约束说明

无。