1 | #pragma once |
---|---|
2 | |
3 | #if !defined(_MSC_VER) && __cplusplus < 201402L |
4 | #error C++14 or later compatible compiler is required to use PyTorch. |
5 | #endif |
6 | |
7 | #include <torch/autograd.h> |
8 | #include <torch/cuda.h> |
9 | #include <torch/data.h> |
10 | #include <torch/enum.h> |
11 | #include <torch/fft.h> |
12 | #include <torch/jit.h> |
13 | #include <torch/linalg.h> |
14 | #include <torch/nested.h> |
15 | #include <torch/nn.h> |
16 | #include <torch/optim.h> |
17 | #include <torch/serialize.h> |
18 | #include <torch/sparse.h> |
19 | #include <torch/special.h> |
20 | #include <torch/types.h> |
21 | #include <torch/utils.h> |
22 | #include <torch/version.h> |
23 |