我想这个问题很模糊，但我有一个刚刚构建的 PyTorch 模型，它无法转换为 CoreML，因为 ONNX 添加了Gather操作。完整的模型实际上是两个独立模型的合并，旨在通过尽可能长时间地在 GPU/Metal 上进行处理来提高性能。
构建这个“复合”模型需要我创建几个切片，形式为x = y[:, 0]，我想知道这些是否可能是Gather操作的原因？
我确实意识到我可以创建一个自定义层，但是我刚刚经历了 CoreML 中自定义层的可怕惨败，这浪费了很多很多时间，而且让我无处可去，所以我试图找到另一种解决问题的方法.
如果找到绕过这些切片的方法会阻止 ONNX 添加Gather，我愿意寻找解决方案。
任何想法表示赞赏。