Получение ответа из нейросети

в c++ проекте подключена библиотека libtorch. Для неё использую отскриптованную и обученную pytorch модель. На единичках ответы сходятся и в python и в c++, но в с++ не могу "вытащить" из нее правильные ответы.

    LoadModel();

    auto inputs = torch::from_blob(imgData, { 1, 1, 28, 28 });
    auto output = model.forward({ inputs }).toTensor().cpu();

    int class_index = std::get<1>(output.max(1)).item().toInt();

    return class_index;

В качестве аргумента функции приходит double массив размером 784. На все цифры реагирует примерно одинаково. Подскажите как правильно получить правильные ответы и, если кто знает, литературу по torch в c++ с нормальными примерами, а то все в питоне и разбираться сложно.