跳到主要内容

GetTensor

函数功能​

获取tensor类型的属性值。

函数原型​

const Tensor *GetTensor(const size_t index) const

参数说明​

参数输入/输出说明
index输入属性在IR原型定义中以及在OP_IMPL注册中的索引。

返回值​

指向属性值的指针。

约束说明​

无

调用示例​

const RuntimeAttrs * runtime_attrs = kernel_context->GetAttrs();
const Tensor *attr0 = runtime_attrs->GetTensor(0);