GetTensorShape
函数功能
获取TensorShape类型的属性值。
函数原型
std::shared_ptr<TensorShape> GetTensorShape() const
参数说明
无
返回值说明
std::shared_ptr<TensorShape>,AttrValue的TensorShape类型属性值的指针。
约束说明
无。
调用示例
算子实现时可使用如下接口获取算子A的属性值。
AttrValue *attr = ctx.GetAttr("attr1"); auto shape = attr->GetTensorShape();
父主题: 暂不支持使用接口