tch 0.0.4

Rust wrappers for the PyTorch C++ api (libtorch).
Documentation 
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

/// A small neural-network library based on Torch.
///
/// This library tries to stay as close as possible to the original
/// Python and C++ implementations.
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 rnn;
pub use rnn::LSTM;

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

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

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