Download
|
User Forum
|
Contact Us
|
Bug Reports
Home
>
Examples
Loading...
Searching...
No Matches
extended
parameterisations
Par04
include
Par04TorchInference.hh
Go to the documentation of this file.
1
//
2
// ********************************************************************
3
// * License and Disclaimer *
4
// * *
5
// * The Geant4 software is copyright of the Copyright Holders of *
6
// * the Geant4 Collaboration. It is provided under the terms and *
7
// * conditions of the Geant4 Software License, included in the file *
8
// * LICENSE and available at http://cern.ch/geant4/license . These *
9
// * include a list of copyright holders. *
10
// * *
11
// * Neither the authors of this software system, nor their employing *
12
// * institutes,nor the agencies providing financial support for this *
13
// * work make any representation or warranty, express or implied, *
14
// * regarding this software system or assume any liability for its *
15
// * use. Please see the license in the file LICENSE and URL above *
16
// * for the full disclaimer and the limitation of liability. *
17
// * *
18
// * This code implementation is the result of the scientific and *
19
// * technical work of the GEANT4 collaboration. *
20
// * By using, copying, modifying or distributing the software (or *
21
// * any work based on the software) you agree to acknowledge its *
22
// * use in resulting scientific publications, and indicate your *
23
// * acceptance of all terms of the Geant4 Software license. *
24
// ********************************************************************
25
//
26
27
#ifdef USE_INFERENCE_TORCH
28
#ifndef PAR04TORCHINFERENCE_HH
29
#define PAR04TORCHINFERENCE_HH
30
#include <G4String.hh>
// for G4String
31
#include <G4Types.hh>
// for G4int, G4double
32
#include <memory>
// for unique_ptr
33
#include <vector>
// for vector
34
#include "
Par04InferenceInterface.hh
"
// for Par04InferenceInterface
35
#include <torch/script.h>
36
37
/**
38
* @brief Inference using the TORCH.
39
*
40
* Runs the inference with LibTorch using the input vector from Par04InferenceSetup.
41
*
42
**/
43
44
class
Par04TorchInference :
public
Par04InferenceInterface
45
{
46
public
:
47
Par04TorchInference(
G4String
);
48
Par04TorchInference();
49
50
/// Run inference
51
/// @param[in] aGenVector Input latent space and conditions
52
/// @param[out] aEnergies Model output = generated shower energies
53
/// @param[in] aSize Size of the output
54
void
RunInference(std::vector<float> aGenVector, std::vector<G4double>& aEnergies,
int
aSize);
55
56
private
:
57
torch::jit::script::Module fModule;
58
59
};
60
61
#endif
/* PAR04TORCHINFERENCE_HH */
62
#endif
Par04InferenceInterface.hh
G4String
Par04InferenceInterface
Definition
Doxymodules_parameterisations.h:84
Generated on Tue Jan 30 2024 12:18:45 for Geant4 examples by
1.9.8
Applications
|
User Support
|
Publications
|
Collaboration