（beta）torch_npu.contrib.module.ChannelShuffle

产品支持情况

[object Object]undefined

功能说明

• API功能：应用NPU兼容的通道shuffle操作。

• 等价计算逻辑：

  split_shuffle=False场景可使用[object Object]等价替换[object Object]，两者计算逻辑一致。

  [object Object]

函数原型

参数说明

计算参数

• in_channels ([object Object])：必选参数。输入张量中的通道总数。
• groups ([object Object])：可选参数。shuffle组数。默认值为2。
• split_shuffle ([object Object])：可选参数。shuffle后是否执行chunk操作。默认值为True。

计算输入

• x1 ([object Object])：输入张量。 shape为$(N, C_{in}, *)$。
• x2 ([object Object])：输入张量。 shape为$(N, C_{in}, *)$。

返回值说明

• out1 ([object Object])：输出张量。 shape为$(N, C_{out}, *)$。
• out2 ([object Object])：输出张量。 shape为$(N, C_{out}, *)$。

约束说明

只实现了groups为2场景，请自行修改其他groups场景。

调用示例