Class Acts::CugraphTrackBuilding
-
class CugraphTrackBuilding : public Acts::TrackBuildingBase
Public Functions
-
virtual std::vector<std::vector<int>> operator()(std::any nodes, std::any edges, std::any edge_weights, std::vector<int> &spacepointIDs) override
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 std::vector<std::vector<int>> operator()(std::any nodes, std::any edges, std::any edge_weights, std::vector<int> &spacepointIDs) override