Как насчет
prob = output.probs.detach().cpu().numpy()
Скорее всего, безотносительно соответствий Ваша экосистема разработки. т.е. язык, Вы, вероятно, будете использовать снаружи просто саму энергию, поскольку это имеет полезные вспомогательные библиотеки
Я лично использовал +python в этом trac плагине
http://www.vim.org/scripts/script.php?script_id=2147
после обширного использования и приобретения знаний из этого потрясающего сценария отладчика...
http://www.vim.org/scripts/script.php?script_id=1929
До работы над другими машинами обычно Вы просто живете без материала для небольших заданий. но можно всегда использовать netrw для scp и т.д. в случае необходимости..
но действительно это - то, что Вы хотите изучить и можете использовать