ActsPlugins::GraphConstructionBase Class Reference

#include <ActsPlugins/Gnn/Stages.hpp>

Inheritance diagram for ActsPlugins::GraphConstructionBase:

Public Member Functions
virtual	~GraphConstructionBase ()=default
virtual PipelineTensors	operator() (std::vector< float > &inputValues, std::size_t numNodes, const std::vector< std::uint64_t > &moduleIds, const ExecutionContext &execContext={})=0
	Perform the graph construction.

Constructor & Destructor Documentation

~GraphConstructionBase()

virtual ActsPlugins::GraphConstructionBase::~GraphConstructionBase ( )

virtualdefault

Member Function Documentation

operator()()

virtual PipelineTensors ActsPlugins::GraphConstructionBase::operator() ( std::vector< float > & inputValues, std::size_t numNodes, const std::vector< std::uint64_t > & moduleIds, const ExecutionContext & execContext = {} )

pure virtual

Perform the graph construction.

Parameters

inputValues	Flattened input data
numNodes	Number of nodes. inputValues.size() / numNodes then gives the number of features
moduleIds	Module IDs of the features (used for module-map-like graph construction)
execContext	Device & stream information

Returns

(node_features, edge_features, edge_index)

Implemented in ActsPlugins::ModuleMapCuda, and ActsPlugins::TorchMetricLearning.