在网络中添加一个ShuffleLayer,添加完成后需要调用shuffleLayer的API来指定具体的shuffle操作,否则该layer不执行任何操作,network无法继续组网。
ShuffleLayer *AddShuffle(Tensor *input) noexcept;
参数名 |
输入/输出 |
说明 |
---|---|---|
input |
输入 |
ShuffleLayer的输入张量。 |
返回一个shuffleLayer,如果添加失败时返回nullptr。