SetTensorShape

函数功能

设置TensorShape类型的属性值。

函数原型

bool SetTensorShape(const TensorShape *shape)

参数说明

返回值说明

bool类型。

• true：表示设置成功。
• false：表示设置失败。

约束说明

无。

调用示例

算子实现时可使用如下接口设置算子A的属性值。

AttrValue *attr = ctx.GetAttr("attr1");
bool ret = attr->SetTensorShape(shape);