File Stages.hpp

namespace Acts

Note

This file is foreseen for the Geometry module to replace Extent

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)

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) = 0

Perform the graph construction.

Parameters

inputValues – Flattened input data

Returns

(node_tensor, edge_tensore)

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)