（beta）torch_npu.contrib.module.Focus

产品支持情况

[object Object]undefined

功能说明

使用NPU亲和写法替换YOLOv5中的原生Focus。

函数原型

参数说明

计算参数

• c1 ([object Object])：输入图像中的通道数。
• c2 ([object Object])：卷积产生的通道数。
• k ([object Object])：卷积核大小，默认值为1。
• s ([object Object])：卷积步长，默认值为1。
• p ([object Object])：填充，默认值为None。
• g ([object Object])：从输入通道到输出通道的分组数，默认值为1。
• act ([object Object])：是否使用激活函数，默认值为True。

计算输入

• x([object Object]): 输入张量。

返回值说明

[object Object]

Focus计算结果。

调用示例