def add_shuffle(self: ascendie.Network, arg0: ascendie.Tensor) -> ascendie.ShuffleLayer
函数功能
在网络中添加一个ShuffleLayer,添加完成后需要调用shuffleLayer的API来指定具体的shuffle操作,否则该layer不执行任何操作,network无法继续组网。
函数原型
add_shuffle(self: ascendie.Network, arg0: ascendie.Tensor) -> ascendie.ShuffleLayer
约束说明
- 当前版本只支持静态shape输入,即input的每个维度都大于0。
- 当前不支持动态shape。
参数说明
参数名 |
参数类型 |
输入/输出 |
说明 |
---|---|---|---|
input |
Tensor |
输入 |
ShuffleLayer的输入张量。 |