Class Acts::GraphConstructionBase
-
class GraphConstructionBase
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, int deviceHint = -1) = 0
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)
-
virtual ~GraphConstructionBase() = default