Acts::EdgeClassificationBase Class Reference

#include <acts/blob/sPHENIX/Plugins/ExaTrkX/include/Acts/Plugins/ExaTrkX/Stages.hpp>

Inheritance diagram for Acts::EdgeClassificationBase:

Public Member Functions
virtual std::tuple< std::any, std::any, std::any >	operator() (std::any nodes, std::any edges, int deviceHint=-1)=0
virtual	~EdgeClassificationBase ()=default

Detailed Description

Definition at line 38 of file Stages.hpp.

View newest version in sPHENIX GitHub at line 38 of file Stages.hpp

Constructor & Destructor Documentation

virtual Acts::EdgeClassificationBase::~EdgeClassificationBase ( )

virtualdefault

Member Function Documentation

virtual std::tuple<std::any, std::any, std::any> Acts::EdgeClassificationBase::operator() ( std::any  nodes, std::any  edges, int  deviceHint = -1  )

pure virtual

Perform edge classification

Parameters

nodes	Node tensor with shape (n_nodes, n_node_features)
edges	Edge-index tensor with shape (2, n_edges)
deviceHint	Which GPU to pick. Not relevant for CPU-only builds

Returns

(node_tensor, edge_tensor, score_tensor)

Implemented in Acts::TorchEdgeClassifier, and Acts::OnnxEdgeClassifier.

---

The documentation for this class was generated from the following file:

• acts/blob/sPHENIX/Plugins/ExaTrkX/include/Acts/Plugins/ExaTrkX/Stages.hpp