tch 0.0.1

PyTorch wrappers for rust
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

mod var_store;
pub use var_store::Path;
pub use var_store::VarStore;

mod module;
pub use module::{Module, ModuleT};

mod linear;
pub use linear::Linear;

mod conv2d;
pub use conv2d::{Conv2D, Conv2DConfig};

mod batch_norm;
pub use batch_norm::BatchNorm2D;

mod func;
pub use func::{Func, FuncT};

mod sequential;
pub use sequential::Sequential;
pub use sequential::SequentialT;

mod c_optimizer;
mod optimizer;
pub use optimizer::Optimizer;