File Stages.hpp
-
namespace Acts
Note
This file is foreseen for the
Geometry
module to replaceExtent
-
class EdgeClassificationBase
- #include <Acts/Plugins/ExaTrkX/Stages.hpp>
Subclassed by Acts::OnnxEdgeClassifier, Acts::TorchEdgeClassifier
Public Functions
-
virtual ~EdgeClassificationBase() = default
-
virtual std::tuple<std::any, std::any, std::any> operator()(std::any nodes, std::any edges) = 0
Perform edge classification.
- Parameters
nodes – Node tensor with shape (n_nodes, n_node_features)
edges – Edge-index tensor with shape (2, n_edges)
- Returns
(node_tensor, edge_tensor, score_tensor)
-
virtual ~EdgeClassificationBase() = default
-
class GraphConstructionBase
- #include <Acts/Plugins/ExaTrkX/Stages.hpp>
Subclassed by Acts::OnnxMetricLearning, Acts::TorchMetricLearning
Public Functions
-
virtual ~GraphConstructionBase() = default
-
virtual std::tuple<std::any, std::any> operator()(std::vector<float> &inputValues, std::size_t numNodes) = 0
Perform the graph construction.
- Parameters
inputValues – Flattened input data
numNodes – number of nodes. inputValues.size() / numNodes then gives the number of features
- Returns
(node_tensor, edge_tensore)
-
virtual ~GraphConstructionBase() = default
-
class TrackBuildingBase
- #include <Acts/Plugins/ExaTrkX/Stages.hpp>
Subclassed by Acts::BoostTrackBuilding, Acts::CugraphTrackBuilding
Public Functions
-
virtual ~TrackBuildingBase() = default
-
virtual std::vector<std::vector<int>> operator()(std::any nodes, std::any edges, std::any edgeWeights, std::vector<int> &spacepointIDs) = 0
Perform track building.
- Parameters
nodes – Node tensor with shape (n_nodes, n_node_features)
edges – Edge-index tensor with shape (2, n_edges)
edgeWeights – Edge-weights of the previous edge classification phase
spacepointIDs – IDs of the nodes (must have size=n_nodes)
- Returns
tracks (as vectors of node-IDs)
-
virtual ~TrackBuildingBase() = default
-
class EdgeClassificationBase