Interface for ONNX model inference for PALLAS plasma simulation beam generation. More...

Public Member Functions
	PlasmaMLPALLASOnnxInference (const std::string &modelPath)
	Constructor.

BeamParameters	GenerateBeam (G4double fXof, G4double fA0, G4double fCN2, G4double fPressure)
	Generate beam parameters using the ONNX model.

Detailed Description

Interface for ONNX model inference for PALLAS plasma simulation beam generation.

Class for performing ONNX model inference to generate plasma beam parameters.

Author: Arnaud HUBER huber.nosp@m.@lp2.nosp@m.ib.in.nosp@m.2p3..nosp@m.fr - Alexei SYTOV sytov.nosp@m.@inf.nosp@m.n.it

This class wraps an ONNX runtime session and provides a convenient interface for generating beam parameters given laser and plasma conditions.

Constructor & Destructor Documentation

PlasmaMLPALLASOnnxInference::PlasmaMLPALLASOnnxInference ( const std::string & modelPath )

Constructor.

Parameters

modelPath Path to the ONNX model file

Initializes the ONNX runtime environment and loads the specified model.

Parameters

modelPath Path to the ONNX model file

Initializes the ONNX runtime environment, sets session options, and loads the specified ONNX model. Also initializes input and output node names.

BeamParameters PlasmaMLPALLASOnnxInference::GenerateBeam	(	G4double	fXof,
		G4double	fA0,
		G4double	fCN2,
		G4double	fPressure
	)

Generate beam parameters using the ONNX model.

Generates beam parameters using the ONNX model.

Parameters

Parameters

This method rescales inputs, runs inference on the ONNX model, converts outputs to physical units, and samples transverse positions/momenta.

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