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.

Member Data Documentation

edgeFeatures

std::optional<Tensor<float> > ActsPlugins::PipelineTensors::edgeFeatures

Optional tensor containing edge feature data.

edgeIndex

Tensor<std::int64_t> ActsPlugins::PipelineTensors::edgeIndex

Tensor containing edge connectivity indices.

edgeScores

std::optional<Tensor<float> > ActsPlugins::PipelineTensors::edgeScores

Optional tensor containing edge classification scores.

nodeFeatures

Tensor<float> ActsPlugins::PipelineTensors::nodeFeatures

Tensor containing node feature data.