在网络中添加一个ShuffleLayer,添加完成后需要调用shuffleLayer的API来指定具体的shuffle操作,否则该layer不执行任何操作,network无法继续组网。
add_shuffle(self: ascendie.pyascendie.Network, arg0: ascendie.pyascendie.Tensor) -> AscendIE::ShuffleLayer
参数名 |
参数类型 |
输入/输出 |
说明 |
---|---|---|---|
input |
Tensor |
输入 |
ShuffleLayer的输入张量。 |