List of all items

Structs

• LoadSaveError
• gradients::ClassActivation
• gradients::ClassBias
• gradients::ClassWrapper
• layers::Bias1d
• layers::Conv1d
• layers::Dense
• layers::Diag
• layers::GLU
• layers::LR
• layers::RMSDiv
• layers::Residual
• layers::ScalarScale
• layers::Softmax
• layers::Swish
• misc::ExpAvg
• optimizers::Momentum
• optimizers::RMSProp
• optimizers::TrainInfo
• optimizers::TrainParams
• shapes::Axes2
• shapes::Axes3
• shapes::Axes4
• shapes::Axes5
• shapes::Axes6
• shapes::Axis
• shapes::Const

Enums

• gradients::GradClass
• layers::Activation
• misc::Decay

Traits

• BackpropModule
• Dtype
• Float
• LoadableModule
• Module
• ResetParams
• RevModule
• TracedModule
• Unit
• VisualizableUnit
• gradients::Gradients
• layers::Conv1dKernel
• loss::CosineLoss
• loss::DiffLoss
• loss::LogitLoss
• optimizers::GradAdjuster
• optimizers::GradApplyer
• shapes::Array
• shapes::Axes
• shapes::BroadcastShapeTo
• shapes::BroadcastStridesTo
• shapes::ConstDim
• shapes::ConstShape
• shapes::Dim
• shapes::HasAxes
• shapes::HasShape
• shapes::RealizeShapeTo
• shapes::ReduceShape
• shapes::ReduceShapeTo
• shapes::ReduceStridesTo
• shapes::Shape
• shapes::SliceShape

Functions

• train_batch
• train_batch_parallel
• train_step

Type Aliases

• Error
• shapes::Rank0
• shapes::Rank1
• shapes::Rank2
• shapes::Rank3
• shapes::Rank4
• shapes::Rank5
• shapes::Rank6