文档
注册
评分
提单
论坛
小AI

SetTensorShape

函数功能

设置TensorShape类型的属性值。

函数原型

bool SetTensorShape(const TensorShape *shape)

参数说明

参数

输入/输出

说明

shape

输入

需要设置的TensorShape类型的属性值。

返回值说明

bool类型。

  • true:表示设置成功。
  • false:表示设置失败。

约束说明

无。

调用示例

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

AttrValue *attr = ctx.GetAttr("attr1");
bool ret = attr->SetTensorShape(shape);
搜索结果
找到“0”个结果

当前产品无相关内容

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