Замените IValue Структуры TorchScript В Onnxruntime

#onnxruntime #torchscript

Вопрос:

После того, как я скрыл свою модель pytorch в onnx, мне нужно реализовать модель с помощью onnxruntime. В исходном коде torch::jit есть переменная, инициализированная как значение IV, и я хочу заменить ее структурой, которая имеет ту же функцию в onnxruntime. Кто-нибудь знаком с torchscript и onnxruntime? Код выглядит следующим образом:

 torch::jit::IValue subsampling_cache_;
subsampling_cache_ = std::move(torch::jit::IValue());