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