ShuffleLayer *AddShuffle(Tensor *input) noexcept;

函数功能

在网络中添加一个ShuffleLayer，添加完成后需要调用shuffleLayer的API来指定具体的shuffle操作，否则该layer不执行任何操作，network无法继续组网。

函数原型

ShuffleLayer *AddShuffle(Tensor *input) noexcept;

约束说明

• 当前版本只支持静态shape输入，即input的每个维度都大于0。
• input维度需在[0,8]。

参数说明

返回值说明

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