1#[macro_use]
2extern crate lazy_static;
3
4pub mod data;
5
6mod error;
7pub use error::TchError;
8pub type Result<T> = std::result::Result<T, error::TchError>;
9
10pub(crate) mod wrappers;
11pub use wrappers::device::{Cuda, Device};
12pub use wrappers::jit::{self, CModule, IValue, TrainableCModule};
13pub use wrappers::kind::{self, Kind};
14pub use wrappers::layout::Layout;
15pub use wrappers::optimizer::COptimizer;
16#[cfg(feature = "python-extension")]
17pub use wrappers::python;
18pub use wrappers::scalar::Scalar;
19pub use wrappers::utils;
20pub use wrappers::{
21 get_num_interop_threads, get_num_threads, manual_seed, set_num_interop_threads,
22 set_num_threads, QEngine,
23};
24
25mod tensor;
26pub use tensor::{
27 autocast, display, index, no_grad, no_grad_guard, with_grad, IndexOp, NewAxis, NoGradGuard,
28 Reduction, Shape, Tensor, TensorIndexer,
29};
30
31pub mod nn;
32pub mod vision;