GetOptionalInputTensor

函数功能

基于算子IR原型定义，获取`OPTIONAL_INPUT`类型的输入tensor指针

函数原型

const Tensor *GetOptionalInputTensor(const size_t ir_index) const;

参数说明

返回值说明

指定ir_index的输入tensor指针，当输入index非法或该INPUT没有实例化时，返回空指针。

关于Tensor类型的定义，请参见Tensor类。

调用示例

ge::graphStatus Tiling4ReduceCommon(TilingContext* context) {
  auto in_shape = context->GetInputShape(0);
  GE_ASSERT_NOTNULL(in_shape);
  auto axes_tensor = context->GetOptionalInputTensor(1);
  ...
}