Acts::GraphConstructionBase Class Reference

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

Inheritance diagram for Acts::GraphConstructionBase:

Public Member Functions
virtual std::tuple< std::any, std::any >	operator() (std::vector< float > &inputValues, std::size_t numNodes, int deviceHint=-1)=0
virtual	~GraphConstructionBase ()=default

Detailed Description

Definition at line 21 of file Stages.hpp.

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

Constructor & Destructor Documentation

virtual Acts::GraphConstructionBase::~GraphConstructionBase ( )

virtualdefault

Member Function Documentation

virtual std::tuple<std::any, std::any> Acts::GraphConstructionBase::operator() ( std::vector< float > &  inputValues, std::size_t  numNodes, int  deviceHint = -1  )

pure virtual

Perform the graph construction

Parameters

inputValues	Flattened input data
numNodes	Number of nodes. inputValues.size() / numNodes then gives the number of features
deviceHint	Which GPU to pick. Not relevant for CPU-only builds

Returns

(node_tensor, edge_tensore)

Implemented in Acts::TorchMetricLearning, and Acts::OnnxMetricLearning.

---

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

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