OP_OUTSHAPE
宏功能
针对需要计算结果来确定输出shape的算子，如NonZero算子，该宏用于存放此类算子输出shape的aclTensor。
宏原型
OP_OUTSHAPE（x...）
参数说明
参数 |
输入/输出 |
说明 |
|---|---|---|
x... |
输入 |
包含两部分，第一个参数是存放输出tensor shape的aclTensor，第二个参数是存放需要更新输出shape的tensor索引。 |
约束说明
无
调用示例
1 2 | // 表示算子将输出tensor的shape存放到outShapeTensor中，并且用来更新idx=0的输出tensor的shape OP_OUTSHAPE({outShapeTensor, 0}); |
父主题： 常用宏和类