下载
中文
注册

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的输入张量。