def add_shuffle(self: Network, input: Tensor) -> ShuffleLayer

函数功能

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

函数原型

add_shuffle(self: ascendie.pyascendie.Network, arg0: ascendie.pyascendie.Tensor) -> AscendIE::ShuffleLayer

约束说明

参数说明

参数名

参数类型

输入/输出

说明

input

Tensor

输入

ShuffleLayer的输入张量。