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

SetListTensorShape

函数功能

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

函数原型

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

参数说明

参数

输入/输出

说明

shape

输入

需要设置AttrValue的list TensorShape的属性值。

返回值说明

uint32_t,设置成功的个数。

约束说明

无。

调用示例

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

AttrValue *attr = ctx.GetAttr("attr1");
std::vector<TensorShape *> vals;
vals.push_back(shape);
uint32_t successNum = attr->SetListTensorShape(vals);
搜索结果
找到“0”个结果

当前产品无相关内容

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