carla::learning Namespace Reference

Classes
struct	Inputs
class	NeuralModel
struct	NeuralModelImpl
struct	Outputs
struct	WheelInput
struct	WheelOutput

Functions
torch::jit::IValue	GetWheelTensorInputs (WheelInput &wheel)
WheelOutput	GetWheelTensorOutput (const at::Tensor &particle_forces, const at::Tensor &wheel_forces)
WheelOutput	GetWheelTensorOutputDynamic (const at::Tensor &particle_forces, const at::Tensor &wheel_forces)
void	test_learning ()

Function Documentation

GetWheelTensorInputs()

torch::jit::IValue carla::learning::GetWheelTensorInputs

(

WheelInput &

wheel

)

Definition at line 36 of file pytorch.cpp.

References 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.

Referenced by carla::learning::NeuralModel::Forward(), and carla::learning::NeuralModel::ForwardDynamic().

GetWheelTensorOutput()

WheelOutput carla::learning::GetWheelTensorOutput	(	const at::Tensor &	particle_forces,
		const at::Tensor &	wheel_forces
	)

Definition at line 67 of file pytorch.cpp.

References carla::learning::WheelOutput::_particle_forces, carla::learning::WheelOutput::wheel_forces_x, carla::learning::WheelOutput::wheel_forces_y, carla::learning::WheelOutput::wheel_forces_z, carla::learning::WheelOutput::wheel_torque_x, carla::learning::WheelOutput::wheel_torque_y, and carla::learning::WheelOutput::wheel_torque_z.

Referenced by carla::learning::NeuralModel::Forward(), and carla::learning::NeuralModel::ForwardCUDATensors().

GetWheelTensorOutputDynamic()

WheelOutput carla::learning::GetWheelTensorOutputDynamic	(	const at::Tensor &	particle_forces,
		const at::Tensor &	wheel_forces
	)

Definition at line 93 of file pytorch.cpp.

References carla::learning::WheelOutput::_particle_forces, carla::learning::WheelOutput::wheel_forces_x, carla::learning::WheelOutput::wheel_forces_y, and carla::learning::WheelOutput::wheel_forces_z.

Referenced by carla::learning::NeuralModel::ForwardDynamic().

test_learning()

void carla::learning::test_learning

(

)

Definition at line 27 of file pytorch.cpp.