File OnnxMetricLearning.hpp

namespace Acts

Note

This file is foreseen for the Geometry module to replace Extent

class OnnxMetricLearning : public Acts::GraphConstructionBase

#include <Acts/Plugins/ExaTrkX/OnnxMetricLearning.hpp>

Public Functions

OnnxMetricLearning(const Config &cfg, std::unique_ptr<const Logger> logger)

~OnnxMetricLearning()

inline Config config() const

virtual std::tuple<std::any, std::any> operator()(std::vector<float> &inputValues) override

Perform the graph construction.

Parameters

inputValues – Flattened input data

Returns

(node_tensor, edge_tensore)

Private Functions

void buildEdgesWrapper(std::vector<float> &embedFeatures, std::vector<int64_t> &edgeList, int64_t numSpacepoints, const Logger &logger) const

inline const auto &logger() const

Private Members

Config m_cfg

std::unique_ptr<Ort::Env> m_env

std::unique_ptr<const Acts::Logger> m_logger

std::unique_ptr<Ort::Session> m_model

struct Config

#include <Acts/Plugins/ExaTrkX/OnnxMetricLearning.hpp>

Public Members

int embeddingDim = 8

int knnVal = 500

std::string modelPath

float rVal = 1.6

int spacepointFeatures = 3

namespace Ort