List of all items
Structs
- dtypes::AMP
- nn::builders::Abs
- nn::builders::AccurateGeLU
- nn::builders::AddInto
- nn::builders::AvgPool2D
- nn::builders::AvgPoolGlobal
- nn::builders::BatchNorm1D
- nn::builders::BatchNorm2D
- nn::builders::Bias2D
- nn::builders::Conv2D
- nn::builders::ConvTrans2D
- nn::builders::Cos
- nn::builders::Dropout
- nn::builders::DropoutOneIn
- nn::builders::Embedding
- nn::builders::Exp
- nn::builders::FastGeLU
- nn::builders::Flatten2D
- nn::builders::GeLU
- nn::builders::GeneralizedResidual
- nn::builders::LayerNorm1D
- nn::builders::LeakyReLU
- nn::builders::Linear
- nn::builders::Ln
- nn::builders::LogSoftmax
- nn::builders::MaxPool2D
- nn::builders::MaxPoolGlobal
- nn::builders::MinPool2D
- nn::builders::MinPoolGlobal
- nn::builders::MultiHeadAttention
- nn::builders::ReLU
- nn::builders::Repeated
- nn::builders::Reshape
- nn::builders::Residual
- nn::builders::Sigmoid
- nn::builders::Sin
- nn::builders::Softmax
- nn::builders::SplitInto
- nn::builders::Sqrt
- nn::builders::Square
- nn::builders::Tanh
- nn::builders::Transformer
- nn::builders::TransformerDecoder
- nn::builders::TransformerDecoderBlock
- nn::builders::TransformerEncoder
- nn::builders::TransformerEncoderBlock
- nn::builders::UnbiasedLinear
- nn::builders::Upscale2D
- nn::builders::Upscale2DBy
- nn::modules::Abs
- nn::modules::AccurateGeLU
- nn::modules::AddInto
- nn::modules::AvgPool2D
- nn::modules::AvgPoolGlobal
- nn::modules::BatchNorm1D
- nn::modules::BatchNorm2D
- nn::modules::Bias2D
- nn::modules::Conv2D
- nn::modules::ConvTrans2D
- nn::modules::Cos
- nn::modules::Dropout
- nn::modules::DropoutOneIn
- nn::modules::Embedding
- nn::modules::Exp
- nn::modules::FastGeLU
- nn::modules::Flatten2D
- nn::modules::GeLU
- nn::modules::GeneralizedResidual
- nn::modules::LayerNorm1D
- nn::modules::LeakyReLU
- nn::modules::Linear
- nn::modules::Ln
- nn::modules::LogSoftmax
- nn::modules::MaxPool2D
- nn::modules::MaxPoolGlobal
- nn::modules::MinPool2D
- nn::modules::MinPoolGlobal
- nn::modules::MultiHeadAttention
- nn::modules::ReLU
- nn::modules::Repeated
- nn::modules::Residual
- nn::modules::Sigmoid
- nn::modules::Sin
- nn::modules::Softmax
- nn::modules::SplitInto
- nn::modules::Sqrt
- nn::modules::Square
- nn::modules::Tanh
- nn::modules::Transformer
- nn::modules::TransformerDecoder
- nn::modules::TransformerDecoderBlock
- nn::modules::TransformerEncoderBlock
- nn::modules::UnbiasedLinear
- nn::modules::Upscale2D
- nn::modules::Upscale2DBy
- nn::prelu::PReLU
- nn::prelu::PReLU1D
- nn::prelu::builder::PReLU
- nn::prelu::builder::PReLU1D
- nn::tensor_collection::ModuleField
- nn::tensor_collection::RecursiveWalker
- nn::tensor_collection::ScalarField
- nn::tensor_collection::ScalarOptions
- nn::tensor_collection::TensorField
- nn::tensor_collection::TensorOptions
- optim::Adam
- optim::RMSprop
- optim::Sgd
- optim::UnusedTensors
- shapes::Axes2
- shapes::Axes3
- shapes::Axes4
- shapes::Axes5
- shapes::Axes6
- shapes::Axis
- shapes::Const
- tensor::Cpu
- tensor::Cuda
- tensor::Gradients
- tensor::NoneTape
- tensor::OwnedTape
- tensor::Tensor
- tensor::UniqueId
- tensor_ops::AdamConfig
- tensor_ops::Bilinear
- tensor_ops::NearestNeighbor
- tensor_ops::RMSpropConfig
- tensor_ops::SgdConfig
Enums
- nn::tensor_collection::ViewTensorMut
- nn::tensor_collection::ViewTensorName
- nn::tensor_collection::ViewTensorRef
- optim::OptimizerUpdateError
- tensor::CpuError
- tensor::CudaError
- tensor::safetensors::Error
- tensor_ops::Momentum
- tensor_ops::Pool2DKind
- tensor_ops::WeightDecay
Traits
- data::Arange
- data::Collate
- data::ExactSizeDataset
- data::IteratorBatchExt
- data::IteratorCollateExt
- data::IteratorStackExt
- data::OneHotEncode
- dtypes::Dtype
- dtypes::HasDtype
- dtypes::HasUnitType
- dtypes::NotMixedPrecision
- dtypes::SafeZeros
- dtypes::Unit
- nn::BuildModule
- nn::BuildOnDevice
- nn::DeviceBuildExt
- nn::LoadFromNpz
- nn::LoadFromSafetensors
- nn::ModelEMA
- nn::Module
- nn::ModuleMut
- nn::NonMutableModule
- nn::NumParams
- nn::ResetParams
- nn::SaveToNpz
- nn::SaveToSafetensors
- nn::ToDevice
- nn::ToDtype
- nn::ZeroGrads
- nn::ZeroSizedModule
- nn::tensor_collection::ModuleFields
- nn::tensor_collection::ModuleVisitor
- nn::tensor_collection::TensorCollection
- nn::tensor_collection::TensorViewer
- nn::tensor_collection::TensorVisitor
- optim::Optimizer
- shapes::Array
- shapes::AssertSameNumel
- shapes::Axes
- shapes::BroadcastShapeTo
- shapes::BroadcastStridesTo
- shapes::ConstDim
- shapes::ConstShape
- shapes::Dim
- shapes::HasAxes
- shapes::HasShape
- shapes::PermuteShapeTo
- shapes::PermuteStridesTo
- shapes::RealizeShapeTo
- shapes::ReduceShape
- shapes::ReduceShapeTo
- shapes::ReduceStridesTo
- shapes::RemoveDimTo
- shapes::ReplaceDimTo
- shapes::Shape
- shapes::SliceShape
- tensor::AsArray
- tensor::Cache
- tensor::CopySlice
- tensor::HasErr
- tensor::Merge
- tensor::NumpyDtype
- tensor::OnesTensor
- tensor::PutTape
- tensor::RandomU64
- tensor::SampleTensor
- tensor::SplitTape
- tensor::Storage
- tensor::Synchronize
- tensor::Tape
- tensor::TensorFrom
- tensor::TensorFromVec
- tensor::TensorToArray
- tensor::Trace
- tensor::TriangleTensor
- tensor::WithEmptyTape
- tensor::ZerosTensor
- tensor::safetensors::SafeDtype
- tensor_ops::Backward
- tensor_ops::BroadcastTo
- tensor_ops::ChooseFrom
- tensor_ops::Device
- tensor_ops::GatherTo
- tensor_ops::GenericUpscale2D
- tensor_ops::LogSumExpTo
- tensor_ops::MaxTo
- tensor_ops::MeanTo
- tensor_ops::MinTo
- tensor_ops::PermuteTo
- tensor_ops::RealizeTo
- tensor_ops::ReshapeTo
- tensor_ops::Roll
- tensor_ops::SelectTo
- tensor_ops::StddevTo
- tensor_ops::SumTo
- tensor_ops::TryAdd
- tensor_ops::TryAttentionReshape
- tensor_ops::TryConcat
- tensor_ops::TryConcatAlong
- tensor_ops::TryConv2D
- tensor_ops::TryConvTrans2D
- tensor_ops::TryDiv
- tensor_ops::TryEq
- tensor_ops::TryGe
- tensor_ops::TryGt
- tensor_ops::TryLe
- tensor_ops::TryLt
- tensor_ops::TryMatMul
- tensor_ops::TryMul
- tensor_ops::TryNe
- tensor_ops::TryPReLU
- tensor_ops::TryPool2D
- tensor_ops::TryStack
- tensor_ops::TrySub
- tensor_ops::TryUpscale2D
- tensor_ops::UpscaleMethod
- tensor_ops::VarTo
Functions
- flush_denormals_to_zero
- keep_denormals
- losses::binary_cross_entropy_with_logits_loss
- losses::cross_entropy_with_logits_loss
- losses::huber_loss
- losses::kl_div_with_logits_loss
- losses::mae_loss
- losses::mse_loss
- losses::rmse_loss
- losses::smooth_l1_loss
- tensor_ops::abs
- tensor_ops::accurate_gelu
- tensor_ops::add
- tensor_ops::axpy
- tensor_ops::bce_with_logits
- tensor_ops::bool_and
- tensor_ops::bool_not
- tensor_ops::bool_or
- tensor_ops::bool_xor
- tensor_ops::clamp
- tensor_ops::cos
- tensor_ops::div
- tensor_ops::dropout
- tensor_ops::eq
- tensor_ops::exp
- tensor_ops::fast_gelu
- tensor_ops::ge
- tensor_ops::gelu
- tensor_ops::gt
- tensor_ops::huber_error
- tensor_ops::le
- tensor_ops::leakyrelu
- tensor_ops::ln
- tensor_ops::log_softmax
- tensor_ops::lower_tri
- tensor_ops::lt
- tensor_ops::matmul
- tensor_ops::maximum
- tensor_ops::minimum
- tensor_ops::mul
- tensor_ops::nans_to
- tensor_ops::ne
- tensor_ops::negate
- tensor_ops::normalize
- tensor_ops::powf
- tensor_ops::powi
- tensor_ops::prelu
- tensor_ops::recip
- tensor_ops::relu
- tensor_ops::sigmoid
- tensor_ops::sin
- tensor_ops::slice
- tensor_ops::softmax
- tensor_ops::sqrt
- tensor_ops::square
- tensor_ops::sub
- tensor_ops::tanh
- tensor_ops::to_dtype
- tensor_ops::upper_tri