Сейчас я пытаюсь экспортировать модель в ONNX и запустить ее, используя внешнюю библиотеку, которая анализирует ONNX. К сожалению, моя модель использует index_add и masked_scatter_ , и внешняя библиотека не может анализировать операции ONNX, в которые эти два преобразованы (я считаю, что index_add даже не имеет эквивалента ONNX для opset 11 это то, что я использую).
Можно ли как-нибудь переписать эти операции в «более простые» операции, совместимые с ONNX, без значительного снижения производительности (т.е. без необходимости явно писать циклы for в python)? Некоторое время я пытался, но не смог найти эквивалентного способа express этих двоих.