carla::learning::NeuralModel Class Reference

#include <pytorch.h>

Public Member Functions
void	Forward ()
void	ForwardCUDATensors ()
void	ForwardDynamic ()
Outputs &	GetOutputs ()
void	LoadModel (char *filename, int device)
	NeuralModel ()
void	SetInputs (Inputs input)
	~NeuralModel ()

Private Attributes
Inputs	_input
Outputs	_output
std::unique_ptr< NeuralModelImpl >	Model

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.

References Model.

~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 _input, _output, carla::learning::Inputs::braking, carla::learning::GetWheelTensorInputs(), carla::learning::GetWheelTensorOutput(), Model, carla::learning::Inputs::steering, carla::learning::Inputs::terrain_type, carla::learning::Inputs::throttle, carla::learning::Inputs::verbose, carla::learning::Inputs::wheel0, carla::learning::Outputs::wheel0, carla::learning::Inputs::wheel1, carla::learning::Outputs::wheel1, carla::learning::Inputs::wheel2, carla::learning::Outputs::wheel2, carla::learning::Inputs::wheel3, and carla::learning::Outputs::wheel3.

ForwardCUDATensors()

void carla::learning::NeuralModel::ForwardCUDATensors

(

)

Definition at line 253 of file pytorch.cpp.

References _input, _output, carla::learning::Inputs::braking, carla::learning::GetWheelTensorOutput(), Model, carla::learning::Inputs::steering, carla::learning::Inputs::terrain_type, carla::learning::Inputs::throttle, carla::learning::Inputs::verbose, carla::learning::Inputs::wheel0, carla::learning::Outputs::wheel0, carla::learning::Inputs::wheel1, carla::learning::Outputs::wheel1, carla::learning::Inputs::wheel2, carla::learning::Outputs::wheel2, carla::learning::Inputs::wheel3, and carla::learning::Outputs::wheel3.

ForwardDynamic()

void carla::learning::NeuralModel::ForwardDynamic

(

)

Definition at line 214 of file pytorch.cpp.

References _input, _output, carla::learning::Inputs::braking, carla::learning::GetWheelTensorInputs(), carla::learning::GetWheelTensorOutputDynamic(), Model, carla::learning::Inputs::steering, carla::learning::Inputs::terrain_type, carla::learning::Inputs::throttle, carla::learning::Inputs::verbose, carla::learning::Inputs::wheel0, carla::learning::Outputs::wheel0, carla::learning::Inputs::wheel1, carla::learning::Outputs::wheel1, carla::learning::Inputs::wheel2, carla::learning::Outputs::wheel2, carla::learning::Inputs::wheel3, and carla::learning::Outputs::wheel3.

GetOutputs()

Outputs & carla::learning::NeuralModel::GetOutputs

(

)

Definition at line 286 of file pytorch.cpp.

References _output.

LoadModel()

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

Definition at line 162 of file pytorch.cpp.

References Model.

SetInputs()

void carla::learning::NeuralModel::SetInputs

(

Inputs

input

)

Definition at line 177 of file pytorch.cpp.

References _input.

Member Data Documentation

_input

Inputs carla::learning::NeuralModel::_input

private

Definition at line 77 of file pytorch.h.

Referenced by Forward(), ForwardCUDATensors(), ForwardDynamic(), and SetInputs().

_output

Outputs carla::learning::NeuralModel::_output

private

Definition at line 78 of file pytorch.h.

Referenced by Forward(), ForwardCUDATensors(), ForwardDynamic(), and GetOutputs().

Model

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

private

Definition at line 76 of file pytorch.h.

Referenced by Forward(), ForwardCUDATensors(), ForwardDynamic(), LoadModel(), and NeuralModel().

---

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

• pytorch.h
• pytorch.cpp