AddListTensorShape

函数功能

向AttrValue的list TensorShape属性中添加TensorShape类型的元素。

函数原型

std::shared_ptr<TensorShape> AddListTensorShape()

参数说明

无

返回值说明

std::shared_ptr<TensorShape>，向AttrValue的list TensorShape属性中添加的TensorShape元素的指针。

约束说明

无。

调用示例

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

AttrValue *attr = ctx.GetAttr("attr1");
auto shape = attr->AddListTensorShape();