AddListTensor

Description

Adds a Tensor element to the attribute, Tensor list.

Prototype

std::shared_ptr<Tensor> AddListTensor()

Parameters

None

Returns

std::shared_ptr<Tensor> for the pointer to the tensor added to the list of tensors.

Restrictions

None

Example

The following gives the prototype definition of an operator whose OpType is A, with attribute name set to attr1 and type ListTensor.

For details about the supported data types of the attribute in the .ATTR API, see API Description.

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

In this case, you can use the following API during operator implementation to add an attribute value of type tensor for operator A.

AttrValue *attr = ctx.GetAttr("attr1");
auto tensor = attr->AddListTensor();