OP_OUTSHAPE
Function Usage
Stores the aclTensor of the output shape of an operator, for example, the NonZero operator.
Prototype
OP_OUTSHAPE(x...)
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
x... |
Input |
The first parameter is the aclTensor tensor that stores the output tensor shape, and the second parameter is the index of the tensor whose output shape needs to be updated. |
Constraints
Only one output tensor shape (outShapeTensor) can be updated with a shape of (9, ). The shape of each output tensor occupies 9 bits. The first bit indicates the number of dimensions, and the remaining 8 bits indicate the value of each dimension.
Example
1 2 | // The operator stores the shape of the output tensor in outShapeTensor to update the shape of the output tensor whose idx is 0. OP_OUTSHAPE({outShapeTensor, 0}); |
Parent topic: Common Macros and Classes