1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
#[macro_use] extern crate lazy_static; extern crate libc; extern crate thiserror; extern crate zip; pub mod data; mod error; pub use error::TchError; pub(crate) mod wrappers; pub use wrappers::device::{Cuda, Device}; pub use wrappers::jit::{self, CModule, IValue, TrainableCModule}; pub use wrappers::kind::{self, Kind}; pub use wrappers::scalar::Scalar; pub use wrappers::{ get_num_interop_threads, get_num_threads, manual_seed, set_num_interop_threads, set_num_threads, QEngine, }; mod tensor; pub use tensor::{ autocast, index, no_grad, no_grad_guard, with_grad, IndexOp, NewAxis, NoGradGuard, Reduction, Shape, Tensor, TensorIndexer, }; pub mod nn; pub mod vision; pub fn maybe_init_cuda() { unsafe { torch_sys::dummy_cuda_dependency(); } }