Public Member Functions | Public Attributes | List of all members
carla::learning::NeuralModelImpl Struct Reference
+ Collaboration diagram for carla::learning::NeuralModelImpl:

Public Member Functions

torch::jit::IValue GetWheelTensorInputsCUDA (WheelInput &wheel, int wheel_idx)
 NeuralModelImpl ()
 ~NeuralModelImpl ()

Public Attributes

torch::jit::script::Module module
std::vector< at::Tensor > particles_position_tensors
std::vector< at::Tensor > particles_velocity_tensors

Detailed Description

Definition at line 117 of file pytorch.cpp.

Constructor & Destructor Documentation

◆ NeuralModelImpl()

carla::learning::NeuralModelImpl::NeuralModelImpl ( )

Definition at line 119 of file pytorch.cpp.

◆ ~NeuralModelImpl()

carla::learning::NeuralModelImpl::~NeuralModelImpl ( )

Definition at line 121 of file pytorch.cpp.

Member Function Documentation

◆ GetWheelTensorInputsCUDA()

torch::jit::IValue carla::learning::NeuralModelImpl::GetWheelTensorInputsCUDA ( WheelInput wheel,
int  wheel_idx 

Member Data Documentation

◆ module

torch::jit::script::Module carla::learning::NeuralModelImpl::module

Definition at line 120 of file pytorch.cpp.

◆ particles_position_tensors

std::vector<at::Tensor> carla::learning::NeuralModelImpl::particles_position_tensors

Definition at line 122 of file pytorch.cpp.

◆ particles_velocity_tensors

std::vector<at::Tensor> carla::learning::NeuralModelImpl::particles_velocity_tensors

Definition at line 123 of file pytorch.cpp.

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