OUTPUT
函数原型
OUTPUT (x, t)
功能说明
注册算子输出信息。
注册算子输出信息成功后,自动生成算子输出的相关接口,用户获取算子输出的名称、获取算子输出的描述、设置算子输出的描述。
例如,注册算子输出y,算子输出接收的数据类型为TensorType{DT_FLOAT},可调用OUTPUT(y, TensorType{DT_FLOAT})接口,注册算子输出成功后,自动生成以下相关接口
static const string name_out_y();// 返回输出的名称,即“y” TensorDesc get_output_desc_y();// 返回输出y对应的描述 graphStatus update_output_desc_y(const TensorDesc& tensorDesc); );// 设置输出y对应的描述,包括Shape、DataType、Format等信息
参数说明
参数名 |
输入/输出 |
类型 |
描述 |
---|---|---|---|
x |
输入 |
- |
宏参数,算子输出的名称。 |
t |
输入 |
- |
算子输出接收的数据类型,可以是TensorType定义的一个或多个,如果多个,通过“,”隔离,例如: TensorType{DT_FLOAT} TensorType({DT_FLOAT, DT_INT8} |
返回值
无。
异常处理
无。
约束说明
对于同一个算子,注册的算子输出名称需保持唯一,不能重复。