文档管理中心

GetRequiredInputTensor

函数功能

根据算子原型定义中的输入索引获取对应的必选输入tensor指针。

函数原型

收起
自动换行
深色代码主题
复制
  1. const Tensor *GetRequiredInputTensor(const size_t ir_index) const;

参数说明

展开
参数 输入/输出 说明
ir_index 输入 必选输入在算子IR原型定义中的索引,从0开始计数。

返回值

指定的输入tensor指针,当输入ir_index非法时,返回空指针。

关于Tensor类型的定义,请参见Tensor

约束说明

如果输入没有被设置为数据依赖,调用此接口获取tensor时,只能在tensor中获取到正确的shape、format、datatype信息。无法获取到真实的tensor数据地址(获取到的地址为nullptr)。

调用示例

收起
自动换行
深色代码主题
复制
  1. ge::graphStatus InferShape4ConcatD(TilingContext* context) {
  2. auto in_tensor = context->GetRequiredInputTensor(0U);
  3. // ...
  4. }
在 开发与测试 开放能力API 中进行搜索
请输入您想要搜索的关键词