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 ( )

inline

Definition at line 119 of file pytorch.cpp.

~NeuralModelImpl()

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

inline

Definition at line 121 of file pytorch.cpp.

Member Function Documentation

GetWheelTensorInputsCUDA()

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

Definition at line 126 of file pytorch.cpp.

References carla::learning::WheelInput::num_particles, carla::learning::WheelInput::particles_positions, carla::learning::WheelInput::particles_velocities, carla::learning::WheelInput::wheel_angular_velocity, carla::learning::WheelInput::wheel_linear_velocity, carla::learning::WheelInput::wheel_oritentation, and carla::learning::WheelInput::wheel_positions.

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:

• pytorch.cpp