在网络中添加一个ShapeLayer，用来获取输入tensor的shape，shapeLayer的输出tensor是一个1维的int32_t 类型的tensor。
ShapeLayer *AddShape(Tensor *input) noexcept;
参数名 |
输入/输出 |
说明 |
|---|---|---|
input |
输入 |
ShapeLayer的输入张量。 |
返回一个ShapeLayer，如果添加失败时返回nullptr。