【libtorch】——模型

本文主要是介绍【libtorch】——模型，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

libtorch

1. prim ops是什么

for (const auto &node : graph->nodes()) {const auto& kind = node->kind();bool isOutputNode = false;for (const auto output : node->outputs()) {isOutputNode |= std::find(netT->outputName.begin(), netT->outputName.end(), output->debugName()) != netT->outputName.end();}// python prim opsif (!isOutputNode && kind.is_prim() && scope->dealPrime(node)) {continue;}

这篇关于【libtorch】——模型的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---