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 () |
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().
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().
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().
void carla::learning::test_learning | ( | ) |
Definition at line 27 of file pytorch.cpp.
Referenced by FSparseHighDetailMap::SaveMap().