SetListTensor

函数功能

设置list Tensor类型的属性值。

函数原型

uint32_t SetListTensor(const std::vector<Tensor *> &tensor)

参数说明

返回值说明

uint32_t，设置成功的个数。

约束说明

无。

调用示例

例如，OpType为A的算子的原型定义如下所示，属性的名称为attr1，类型为ListTensor。

.ATTR接口中属性类型的取值范围可参见接口说明。

REG_OP(A)
    .INPUT(x, TensorType({ DT_FLOAT }))
    .OUTPUT(y, TensorType({ DT_FLOAT }))
    .ATTR(attr1, ListTensor, {})
    .OP_END_FACTORY_REG(A)

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

std::vector<Tensor *> vals;
vals.push_back(tensor);
uint32_t successNum = attr->SetListTensor(vals);