ActsPlugins::PipelineTensors Struct Reference

Struct that ties together the tensors used in the GNN pipeline. More...

#include <ActsPlugins/Gnn/Stages.hpp>

Collaboration diagram for ActsPlugins::PipelineTensors:

Public Attributes
std::optional< Tensor< float > >	edgeFeatures
	Optional tensor containing edge feature data.
Tensor< std::int64_t >	edgeIndex
	Tensor containing edge connectivity indices.
std::optional< Tensor< float > >	edgeScores
	Optional tensor containing edge classification scores.
Tensor< float >	nodeFeatures
	Tensor containing node feature data.

Detailed Description

Struct that ties together the tensors used in the GNN pipeline.