ShuffleLayer *AddShuffle(Tensor *input) noexcept;

函数功能

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

函数原型

ShuffleLayer *AddShuffle(Tensor *input) noexcept;

约束说明

参数说明

参数名

输入/输出

说明

input

输入

ShuffleLayer的输入张量。

返回值说明

返回一个shuffleLayer,如果添加失败时返回nullptr。