List of all items

Structs

• macros::Activation
• macros::ActivationLayer
• macros::Attention
• macros::AttentionLayer
• macros::Collapse
• macros::CollapseLayer
• macros::Context
• macros::Dataset
• macros::Dense
• macros::DenseLayer
• macros::GELU
• macros::ReLU
• macros::Sigmoid
• macros::Softmax
• macros::SoftmaxLayer
• macros::Swish
• macros::Tanh
• macros::Temporal
• macros::TemporalLayer
• macros::test::TestEval
• nn::tensors::Tensor
• nn::tensors::WithGrad

Enums

• approx::ApproxEquality
• backend::Backend
• macros::ActivationKind
• macros::Backward

Traits

• approx::RelativeEq
• macros::ActivationFn
• macros::Closure
• macros::ClosureMut
• macros::ClosureOnce
• macros::Layer
• nn::tensors::ConstTensorOps
• nn::tensors::Flatten
• nn::tensors::IntoWithGrad
• nn::tensors::StaticShape
• nn::tensors::TensorGrad
• nn::tensors::TensorOps
• prelude::TensorOps

Macros

• prelude::static_model
• static_model

Functions

• approx::approx_eq
• backend::get_backend
• backend::set_backend
• macros::adapt_lr
• macros::asym_distr
• macros::decay_lr
• macros::decay_temp
• macros::test::accuracy_of
• macros::test::percentage_correct
• nn::ops::cpu::cross_entropy_loss
• nn::ops::cpu::cross_entropy_nonzero_loss
• nn::ops::cpu::gelu
• nn::ops::cpu::matmul
• nn::ops::cpu::mse_loss
• nn::ops::cpu::relu
• nn::ops::cpu::sgd
• nn::ops::cpu::sigmoid
• nn::ops::cpu::softmax
• nn::ops::cpu::swish
• nn::ops::cpu::tanh
• nn::ops::dispatch::cross_entropy_loss
• nn::ops::dispatch::cross_entropy_nonzero_loss
• nn::ops::dispatch::gelu
• nn::ops::dispatch::matmul
• nn::ops::dispatch::mse_loss
• nn::ops::dispatch::relu
• nn::ops::dispatch::sgd
• nn::ops::dispatch::sigmoid
• nn::ops::dispatch::softmax
• nn::ops::dispatch::swish
• nn::ops::dispatch::tanh

Type Aliases

• macros::BackFn
• macros::Tensor
• nn::TensorFloat

Constants

• approx::F32_AVG_ERROR
• approx::F32_MAX_ERROR
• approx::F32_MIN_ERROR
• approx::F64_AVG_ERROR
• approx::F64_MAX_ERROR
• approx::F64_MIN_ERROR
• nn::FLOAT_LANES