下载
中文
注册

GetListTensorShape

函数功能

获取AttrValue的list TensorShape类型的属性值。

函数原型

std::vector<TensorShape> GetListTensorShape() const

参数说明

返回值说明

std::vector<TensorShape>,AttrValue的list TensorShape属性值。

约束说明

无。

调用示例、

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

AttrValue *attr = ctx.GetAttr("attr1");
std::vector<TensorShape> shapes = attr->GetListTensorShape();