文档
注册

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();
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词