OP_OUTSHAPE

宏功能

针对需要计算结果来确定输出shape的算子，如NonZero算子，该宏用于存放此类算子输出shape的aclTensor。

宏原型

OP_OUTSHAPE(x...)

参数说明

约束说明

只支持刷新一个输出tensor shape的Tensor（outShapeTensor），其对应shape为(9, )。每个输出tensor的shape占9位，其中第一位表示维度数，剩下8位表示每个维度的具体值。

调用示例

1
2

// 表示算子将输出tensor的shape存放到outShapeTensor中，并且用来更新idx=0的输出tensor的shape
OP_OUTSHAPE({outShapeTensor, 0});