#include "pytorch.h"
#include <torch/torch.h>
#include <torch/script.h>
#include <torchscatter/scatter.h>
#include <torchcluster/cluster.h>
#include <torch/csrc/jit/passes/tensorexpr_fuser.h>
#include <c10/cuda/CUDACachingAllocator.h>
#include <string>
#include <vector>
#include <ostream>
#include <sstream>
类 | |
struct | carla::learning::NeuralModelImpl |
命名空间 | |
namespace | carla |
This file contains definitions of common data structures used in traffic manager. | |
namespace | carla::learning |
宏定义 | |
#define | _GLIBCXX_USE_CXX11_ABI 0 |
函数 | |
void | add_mark (const std::string &text) |
torch::jit::IValue | carla::learning::GetWheelTensorInputs (WheelInput &wheel) |
WheelOutput | carla::learning::GetWheelTensorOutput (const at::Tensor &particle_forces, const at::Tensor &wheel_forces) |
WheelOutput | carla::learning::GetWheelTensorOutputDynamic (const at::Tensor &particle_forces, const at::Tensor &wheel_forces) |
void | carla::learning::test_learning () |
#define _GLIBCXX_USE_CXX11_ABI 0 |
在文件 pytorch.cpp 第 5 行定义.
void add_mark | ( | const std::string & | text | ) |
在文件 pytorch.cpp 第 20 行定义.