CARLA
Public Member Functions | Private Attributes | List of all members
carla::learning::NeuralModel Class Reference

#include <pytorch.h>

+ Collaboration diagram for carla::learning::NeuralModel:

Public Member Functions

void Forward ()
 
void ForwardCUDATensors ()
 
void ForwardDynamic ()
 
OutputsGetOutputs ()
 
void LoadModel (char *filename, int device)
 
 NeuralModel ()
 
void SetInputs (Inputs input)
 
 ~NeuralModel ()
 

Private Attributes

Inputs _input
 
Outputs _output
 
std::unique_ptr< NeuralModelImplModel
 

Detailed Description

Definition at line 61 of file pytorch.h.

Constructor & Destructor Documentation

◆ NeuralModel()

carla::learning::NeuralModel::NeuralModel ( )

Definition at line 159 of file pytorch.cpp.

◆ ~NeuralModel()

carla::learning::NeuralModel::~NeuralModel ( )

Definition at line 290 of file pytorch.cpp.

Member Function Documentation

◆ Forward()

void carla::learning::NeuralModel::Forward ( )

Definition at line 182 of file pytorch.cpp.

References carla::learning::GetWheelTensorInputs(), and carla::learning::GetWheelTensorOutput().

+ Here is the call graph for this function:

◆ ForwardCUDATensors()

void carla::learning::NeuralModel::ForwardCUDATensors ( )

Definition at line 253 of file pytorch.cpp.

References carla::learning::GetWheelTensorOutput().

+ Here is the call graph for this function:

◆ ForwardDynamic()

void carla::learning::NeuralModel::ForwardDynamic ( )

Definition at line 214 of file pytorch.cpp.

References carla::learning::GetWheelTensorInputs(), and carla::learning::GetWheelTensorOutputDynamic().

+ Here is the call graph for this function:

◆ GetOutputs()

Outputs & carla::learning::NeuralModel::GetOutputs ( )

Definition at line 286 of file pytorch.cpp.

◆ LoadModel()

void carla::learning::NeuralModel::LoadModel ( char *  filename,
int  device 
)

Definition at line 162 of file pytorch.cpp.

◆ SetInputs()

void carla::learning::NeuralModel::SetInputs ( Inputs  input)

Definition at line 177 of file pytorch.cpp.

Member Data Documentation

◆ _input

Inputs carla::learning::NeuralModel::_input
private

Definition at line 77 of file pytorch.h.

◆ _output

Outputs carla::learning::NeuralModel::_output
private

Definition at line 78 of file pytorch.h.

◆ Model

std::unique_ptr<NeuralModelImpl> carla::learning::NeuralModel::Model
private

Definition at line 76 of file pytorch.h.


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