Skip to main content

Crate tml_utils

Crate tml_utils 

Source

Re-exports§

pub use autodiff::EvalTape;
pub use autodiff::ExprGraph;
pub use autodiff::Gradients;
pub use autodiff::NodeId;
pub use autodiff::Op;
pub use autodiff::ReverseTape;
pub use autodiff::Tape;
pub use autodiff::TapeError;
pub use autodiff::Var;
pub use data::Sample;
pub use network::Adam;
pub use network::DenseLayer;
pub use network::Flatten;
pub use network::Initializer;
pub use network::KaimingUniform;
pub use network::Layer;
pub use network::LayerDims;
pub use network::LossFunction;
pub use network::MeanSquaredError;
pub use network::ModelBuilder;
pub use network::Optimizer;
pub use network::ReLU;
pub use network::Sequential;
pub use network::Sgd;
pub use network::Sigmoid;
pub use network::TrainConfig;
pub use network::Uniform;
pub use network::XavierUniform;
pub use network::mse_loss;
pub use shape::TensorShape;

Modules§

autodiff
conv
data
network
shape

Macros§

expr
Macro for building differentiable expressions.
shape
tensor

Structs§

Tensor
TensorMut
TensorRef

Type Aliases§

Float