onnx2ncnn中的gather2crop中的疑问

[yyq0210]

detail | 详细描述 | 詳細な説明

tools/onnx/onnx2ncnn.cpp
crop算子节点的axis为何是在gather节点的axis基础上减一？

[dosubot]

Hi @yyq0210! I'm Dosu and I’m helping the ncnn team.

在 onnx2ncnn 的 gather2crop 转换中，crop 节点的 axis 之所以要在 gather 节点的 axis 基础上减一，是因为 ONNX 的张量维度索引从 0 开始，包括 batch 维（N），而 ncnn 默认所有输入都是单张图片（N=1），其 crop 算子的 axis 不包含 batch 维。比如 ONNX 的 [N, C, H, W]，axis=1（C）在 ncnn 里就对应 axis=0（C, H, W）。所以转换时需要做 axis-1 的偏移，保证语义一致。你可以在 onnx2ncnn.cpp 的 fuse_rewrite_gather 函数中看到相关实现和注释，ncnn crop 层的实现也体现了这种维度对应关系（axis=0 通常对应 channel）代码参考crop 层实现。

To reply, just mention @dosu.

---

^{How did I do? Good | Irrelevant | Incorrect | Verbose | Hallucination | Report 🐛 | Other}   

[yyq0210]

谢谢您的回复，感恩！

…

---原始邮件--- 发件人: ***@***.***> 发送时间: 2025年10月5日(周日) 晚上6:03 收件人: ***@***.***>; 抄送: ***@***.******@***.***>; 主题: Re: [Tencent/ncnn] onnx2ncnn中的gather2crop中的疑问 (Issue #6341) dosubot[bot] left a comment (Tencent/ncnn#6341) Hi @yyq0210! I'm Dosu and I’m helping the ncnn team. 在 onnx2ncnn 的 gather2crop 转换中，crop 节点的 axis 之所以要在 gather 节点的 axis 基础上减一，是因为 ONNX 的张量维度索引从 0 开始，包括 batch 维（N），而 ncnn 默认所有输入都是单张图片（N=1），其 crop 算子的 axis 不包含 batch 维。比如 ONNX 的 [N, C, H, W]，axis=1（C）在 ncnn 里就对应 axis=0（C, H, W）。所以转换时需要做 axis-1 的偏移，保证语义一致。你可以在 onnx2ncnn.cpp 的 fuse_rewrite_gather 函数中看到相关实现和注释，ncnn crop 层的实现也体现了这种维度对应关系（axis=0 通常对应 channel）代码参考crop 层实现。 To reply, just mention @dosu. How did I do? Good | Irrelevant | Incorrect | Verbose | Hallucination | Report 🐛 | Other    — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: ***@***.***>

[dosubot]

不客气！很高兴能帮到你，有其他关于 ncnn 或模型转换的问题随时可以 @我。

To reply, just mention @dosu.

---

^{How did I do? Good | Irrelevant | Incorrect | Verbose | Hallucination | Report 🐛 | Other}