File TorchMetricLearning.hpp
-
namespace Acts
Note
This file is foreseen for the
Geometry
module to replaceExtent
-
class TorchMetricLearning : public Acts::GraphConstructionBase
- #include <Acts/Plugins/ExaTrkX/TorchMetricLearning.hpp>
Public Functions
-
~TorchMetricLearning()
-
inline Config config() const
-
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)
Private Functions
-
inline const auto &logger() const
Private Members
-
struct Config
- #include <Acts/Plugins/ExaTrkX/TorchMetricLearning.hpp>
Public Members
-
int embeddingDim = 8
-
int knnVal = 500
-
std::string modelPath
-
int numFeatures = 3
-
float rVal = 1.6
-
bool shuffleDirections = false
-
int embeddingDim = 8
-
~TorchMetricLearning()
-
class TorchMetricLearning : public Acts::GraphConstructionBase
-
namespace c10
-
namespace torch
-
namespace jit
-
namespace jit