Moose::TorchScriptModule Class Reference

#include <TorchScriptModule.h>

Inheritance diagram for Moose::TorchScriptModule:

Public Member Functions
	TorchScriptModule ()
	Construct using a filename which contains the source code in torchscript format. More...
	TorchScriptModule (const std::string &filename)
	Construct using a filename which contains the source code in torchscript format. More...
void	loadNeuralNetwork (const std::string &filename)
	Construct the neural network. More...
virtual torch::Tensor	forward (const torch::Tensor &x) override
	Overriding the forward substitution function for the neural network, unfortunately this cannot be const since it creates a graph in the background. More...

Detailed Description

Definition at line 24 of file TorchScriptModule.h.

Constructor & Destructor Documentation

TorchScriptModule() [1/2]

Moose::TorchScriptModule::TorchScriptModule

(

)

Construct using a filename which contains the source code in torchscript format.

Definition at line 19 of file TorchScriptModule.C.

{}

TorchScriptModule() [2/2]

Moose::TorchScriptModule::TorchScriptModule

(

const std::string &

filename

)

Construct using a filename which contains the source code in torchscript format.

Parameters

filename

The name of the file that contains the neural net

Definition at line 21 of file TorchScriptModule.C.

{ loadNeuralNetwork(filename); }

Member Function Documentation

forward()

torch::Tensor Moose::TorchScriptModule::forward ( const torch::Tensor &  x )

overridevirtual

Overriding the forward substitution function for the neural network, unfortunately this cannot be const since it creates a graph in the background.

Parameters

x	Input tensor for the evaluation

Implements Moose::LibtorchNeuralNetBase.

Definition at line 38 of file TorchScriptModule.C.

{
  std::vector<torch::jit::IValue> inputs(1, x);
  return torch::jit::script::Module::forward(inputs).toTensor();
}

loadNeuralNetwork()

void Moose::TorchScriptModule::loadNeuralNetwork

(

const std::string &

filename

)

Construct the neural network.

Definition at line 24 of file TorchScriptModule.C.

Referenced by TorchScriptModule().

{
  try
  {
    torch::jit::script::Module * base = this;
    *base = torch::jit::load(filename);
  }
  catch (const c10::Error & e)
  {
    mooseError("Error while loading torchscript file ", filename, "!\n", e.msg());
  }
}

---

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

• include/libtorch/utils/TorchScriptModule.h
• src/libtorch/utils/TorchScriptModule.C