Class Acts::TorchMetricLearning
-
class TorchMetricLearning : public Acts::GraphConstructionBase
Public Functions
-
~TorchMetricLearning()
-
virtual std::tuple<std::any, std::any> operator()(std::vector<float> &inputValues, std::size_t numNodes) override
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)
-
struct Config
-
~TorchMetricLearning()