Moose::LibtorchNeuralNetBase Class Reference

This base class is meant to gather the functions and members common in every neural network based on Libtorch. More...

#include <LibtorchNeuralNetBase.h>

Inheritance diagram for Moose::LibtorchNeuralNetBase:

Public Member Functions
virtual	~LibtorchNeuralNetBase ()
virtual torch::Tensor	forward (const torch::Tensor &x)=0

Detailed Description

This base class is meant to gather the functions and members common in every neural network based on Libtorch.

Definition at line 24 of file LibtorchNeuralNetBase.h.

Constructor & Destructor Documentation

~LibtorchNeuralNetBase()

virtual Moose::LibtorchNeuralNetBase::~LibtorchNeuralNetBase ( )

inlinevirtual

Definition at line 28 of file LibtorchNeuralNetBase.h.

{}

Member Function Documentation

forward()

virtual torch::Tensor Moose::LibtorchNeuralNetBase::forward ( const torch::Tensor &  x )

pure virtual

Implemented in Moose::LibtorchArtificialNeuralNet, and Moose::TorchScriptModule.

---

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

• include/libtorch/utils/LibtorchNeuralNetBase.h