ShapeLayer *AddShape(Tensor *input) noexcept;

函数功能

在网络中添加一个ShapeLayer，用来获取输入tensor的shape，shapeLayer的输出tensor是一个1维的int32类型的tensor。

函数原型

ShapeLayer *AddShape(Tensor *input) noexcept;

约束说明

• input维度需在[0, 8]。
• 当前不支持动态shape。

参数说明

返回值说明

返回一个ShapeLayer，如果添加失败则返回nullptr或抛出异常。