Acts::TorchMetricLearning Exception Reference

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

Inheritance diagram for Acts::TorchMetricLearning:

Collaboration diagram for Acts::TorchMetricLearning:

Classes
struct	Config

Public Member Functions
	TorchMetricLearning (const Config &cfg, std::unique_ptr< const Logger > logger)
	~TorchMetricLearning ()
std::tuple< std::any, std::any >	operator() (std::vector< float > &inputValues, std::size_t numNodes, int deviceHint=-1) override
Config	config () const
Public Member Functions inherited from Acts::GraphConstructionBase
virtual	~GraphConstructionBase ()=default

Private Member Functions
const auto &	logger () const

Private Attributes
std::unique_ptr< const Acts::Logger >	m_logger
Config	m_cfg
c10::DeviceType	m_deviceType
std::unique_ptr < torch::jit::Module >	m_model

Detailed Description

Definition at line 26 of file TorchMetricLearning.hpp.

View newest version in sPHENIX GitHub at line 26 of file TorchMetricLearning.hpp

Constructor & Destructor Documentation

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

Definition at line 23 of file TorchMetricLearning.cpp.

View newest version in sPHENIX GitHub at line 23 of file TorchMetricLearning.cpp

References ACTS_DEBUG, ACTS_INFO, Acts::UnitConstants::e, m_cfg, m_deviceType, m_model, and Acts::TorchMetricLearning::Config::modelPath.

Acts::TorchMetricLearning::~TorchMetricLearning

(

)

Definition at line 46 of file TorchMetricLearning.cpp.

View newest version in sPHENIX GitHub at line 46 of file TorchMetricLearning.cpp

Member Function Documentation

Config Acts::TorchMetricLearning::config ( ) const

inline

Definition at line 44 of file TorchMetricLearning.hpp.

View newest version in sPHENIX GitHub at line 44 of file TorchMetricLearning.hpp

References m_cfg().

Here is the call graph for this function:

const auto& Acts::TorchMetricLearning::logger ( ) const

inlineprivate

Definition at line 48 of file TorchMetricLearning.hpp.

View newest version in sPHENIX GitHub at line 48 of file TorchMetricLearning.hpp

References m_logger.

Referenced by operator()().

Here is the caller graph for this function:

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

overridevirtual

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)

Implements Acts::GraphConstructionBase.

Definition at line 48 of file TorchMetricLearning.cpp.

View newest version in sPHENIX GitHub at line 48 of file TorchMetricLearning.cpp

References ACTS_DEBUG, Acts::ACTS_VERBOSE(), i, logger(), m_cfg, m_deviceType, m_model, Acts::TorchMetricLearning::Config::numFeatures, and Acts::detail::vectorToTensor2D().

Here is the call graph for this function:

Member Data Documentation

Config Acts::TorchMetricLearning::m_cfg

private

Definition at line 50 of file TorchMetricLearning.hpp.

View newest version in sPHENIX GitHub at line 50 of file TorchMetricLearning.hpp

Referenced by operator()(), and TorchMetricLearning().

c10::DeviceType Acts::TorchMetricLearning::m_deviceType

private

Definition at line 51 of file TorchMetricLearning.hpp.

View newest version in sPHENIX GitHub at line 51 of file TorchMetricLearning.hpp

Referenced by operator()(), and TorchMetricLearning().

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

private

Definition at line 47 of file TorchMetricLearning.hpp.

View newest version in sPHENIX GitHub at line 47 of file TorchMetricLearning.hpp

std::unique_ptr<torch::jit::Module> Acts::TorchMetricLearning::m_model

private

Definition at line 52 of file TorchMetricLearning.hpp.

View newest version in sPHENIX GitHub at line 52 of file TorchMetricLearning.hpp

Referenced by operator()(), and TorchMetricLearning().

---

The documentation for this exception was generated from the following files:

• acts/blob/sPHENIX/Plugins/ExaTrkX/include/Acts/Plugins/ExaTrkX/TorchMetricLearning.hpp
• acts/blob/sPHENIX/Plugins/ExaTrkX/src/TorchMetricLearning.cpp