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