lar_dl_content::LArDLHelper Class Reference

LArDLHelper class. More...

#include <LArDLHelper.h>

Public Types
typedef torch::jit::script::Module	TorchModel
typedef torch::Tensor	TorchInput
typedef std::vector < torch::jit::IValue >	TorchInputVector
typedef at::Tensor	TorchOutput

Static Public Member Functions
static pandora::StatusCode	LoadModel (const std::string &filename, TorchModel &model)
	Loads a deep learning model. More...
static void	InitialiseInput (const at::IntArrayRef dimensions, TorchInput &tensor)
	Create a torch input tensor. More...
static void	Forward (TorchModel &model, const TorchInputVector &input, TorchOutput &output)
	Run a deep learning model. More...

Detailed Description

LArDLHelper class.

Definition at line 22 of file LArDLHelper.h.

Member Typedef Documentation

typedef torch::Tensor lar_dl_content::LArDLHelper::TorchInput

Definition at line 26 of file LArDLHelper.h.

typedef std::vector<torch::jit::IValue> lar_dl_content::LArDLHelper::TorchInputVector

Definition at line 27 of file LArDLHelper.h.

typedef torch::jit::script::Module lar_dl_content::LArDLHelper::TorchModel

Definition at line 25 of file LArDLHelper.h.

typedef at::Tensor lar_dl_content::LArDLHelper::TorchOutput

Definition at line 28 of file LArDLHelper.h.

Member Function Documentation

void lar_dl_content::LArDLHelper::Forward ( TorchModel &  model, const TorchInputVector &  input, TorchOutput &  output  )

static

Run a deep learning model.

Parameters

model	the model to run
input	the input to run over
output	the tensor to store the output in

Definition at line 41 of file LArDLHelper.cc.

{
    output = model.forward(input).toTensor();
}

void lar_dl_content::LArDLHelper::InitialiseInput ( const at::IntArrayRef  dimensions, TorchInput &  tensor  )

static

Create a torch input tensor.

Parameters

dimensions	the size of each dimension of the tensor: pass as {a, b, c, d} for example
tensor	the tensor to be initialised

Definition at line 34 of file LArDLHelper.cc.

{
    tensor = torch::zeros(dimensions);
}

StatusCode lar_dl_content::LArDLHelper::LoadModel ( const std::string &  filename, LArDLHelper::TorchModel &  model  )

static

Loads a deep learning model.

Parameters

filename	the filename of the model to load
model	the TorchModel in which to store the loaded model

Returns

STATUS_CODE_SUCCESS upon successful loading of the model. STATUS_CODE_FAILURE otherwise.

Definition at line 16 of file LArDLHelper.cc.

{
    try
    {
        model = torch::jit::load(filename);
        std::cout << "Loaded the TorchScript model \'" << filename << "\'" << std::endl;
    }
    catch (const std::exception &e)
    {
        std::cout << "Error loading the TorchScript model \'" << filename << "\':\n" << e.what() << std::endl;
        return STATUS_CODE_FAILURE;
    }

    return STATUS_CODE_SUCCESS;
}

---

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

• LArDLHelper.h
• LArDLHelper.cc