SetListTensorShape

函数功能

设置AttrValue的list TensorShape类型的属性值。

函数原型

uint32_t SetListTensorShape(const std::vector<TensorShape *> &shape)

参数说明

返回值说明

uint32_t，设置成功的个数。

约束说明

无。

调用示例

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

AttrValue *attr = ctx.GetAttr("attr1");
std::vector<TensorShape *> vals;
vals.push_back(shape);
uint32_t successNum = attr->SetListTensorShape(vals);