Плагин ScatterND не найден при преобразовании onnx в модель tensorrt
Возможность конвертировать модель pytorch в onnx и проверять вывод onnx с помощью pytorch. работает нормально. Но когда я пытаюсь преобразовать onnx в тензорный движок, он не работает в парсере onnx и возвращает эту ошибку.
[TensorRT] ОШИБКА: INVALID_ARGUMENT: getPluginCreator не смог найти плагин ScatterND версии 1
Версия Onnx: 1.6.0 Версия Tensorrt: 7.1.3.4
Может ли кто-нибудь помочь здесь, как я могу добавить плагин ScatterND в python и как его использовать? Есть ли другой способ решить эту проблему?
Заранее спасибо.