Неясный вопрос, я полагаю, но у меня есть модель PyTorch, которую я только что построил, и она не может конвертироваться в CoreML, потому что ONNX добавил Gather ops.Полная модель на самом деле представляет собой объединение двух отдельных моделей, предназначенных для повышения производительности за счет максимально возможной длительной обработки на графическом процессоре / металле.
Создание этой "составной" модели потребовало от меня создания парыкусочки, формы x = y[:, 0], и мне интересно, могут ли они быть причиной для Gather ops?
Я понимаю, что могу создать собственный слой, но я только что пережил ужасное фиаско с пользовательскими слоями в CoreML, которое потратило много-много часов и ни к чему меня не привело, поэтому я пытаюсьнайти другой способ решения этой проблемы.
Если поиск обхода этих срезов не позволит ONNX добавить Gather Я бы хотел найти решение.
Любые мысли приветствуются.