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 ops
if (!isOutputNode && kind.is_prim() && scope->dealPrime(node)) {
continue;
}