List of all items
Structs
- Assert
- arrays::AllAxes
- arrays::Axis
- data::SubsetIterator
- devices::Cpu
- devices::PoolAvg
- devices::PoolMax
- devices::PoolMin
- gradients::GradientTape
- gradients::Gradients
- gradients::NoneTape
- gradients::OwnedTape
- gradients::UnusedTensors
- nn::Abs
- nn::AvgPool2D
- nn::AvgPoolGlobal
- nn::BatchNorm2D
- nn::Conv2D
- nn::Cos
- nn::Dropout
- nn::DropoutOneIn
- nn::Exp
- nn::Flatten2D
- nn::GeneralizedResidual
- nn::LayerNorm1D
- nn::Linear
- nn::Ln
- nn::MaxPool2D
- nn::MaxPoolGlobal
- nn::MinPool2D
- nn::MinPoolGlobal
- nn::MultiHeadAttention
- nn::ReLU
- nn::Repeated
- nn::Residual
- nn::Sigmoid
- nn::Sin
- nn::Softmax
- nn::SplitInto
- nn::Sqrt
- nn::Square
- nn::Tanh
- nn::Transformer
- nn::TransformerDecoder
- nn::TransformerDecoderBlock
- nn::TransformerEncoderBlock
- optim::Adam
- optim::AdamConfig
- optim::RMSprop
- optim::RMSpropConfig
- optim::Sgd
- optim::SgdConfig
- optim::UnusedParamsError
- tensor::PhantomTensor
- tensor::Tensor0D
- tensor::Tensor1D
- tensor::Tensor2D
- tensor::Tensor3D
- tensor::Tensor4D
- unique_id::UniqueId
Enums
Traits
- ConstTrue
- arrays::CountElements
- arrays::HasArrayData
- arrays::HasArrayType
- arrays::HasAxes
- arrays::HasLastAxis
- arrays::ZeroElements
- devices::Accumulator
- devices::AllocateZeros
- devices::Device
- devices::DeviceConv2D
- devices::DevicePermute
- devices::DevicePool2D
- devices::DeviceReduce
- devices::DeviceSelect
- devices::FillElements
- devices::ForEachElement
- devices::HasDevice
- devices::MatMul
- devices::MatMulOp
- devices::Transpose
- gradients::CanUpdateWithGradients
- gradients::GradientProvider
- gradients::Merge
- gradients::Tape
- nn::LoadFromNpz
- nn::Module
- nn::ModuleMut
- nn::ResetParams
- nn::SaveToNpz
- numpy::NumpyDtype
- numpy::NumpyShape
- numpy::ReadNumbers
- numpy::WriteNumbers
- optim::Optimizer
- tensor::IntoPhantom
- tensor::IntoTensor
- tensor::PutTape
- tensor::Randomize
- tensor::Tensor
- tensor::TensorCreator
- tensor_ops::BroadcastTo
- tensor_ops::MatMulTrTyping
- tensor_ops::MatMulTyping
- tensor_ops::PermuteTo
- tensor_ops::Reduce
- tensor_ops::ReduceTo
- tensor_ops::Reshape
- tensor_ops::SelectTo
- unique_id::HasUniqueId
Functions
- data::arange
- data::one_hot_encode
- 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
- nn::npz_fread
- nn::npz_fwrite
- numpy::load
- numpy::read
- numpy::save
- numpy::write
- tensor::tensor
- tensor::trace
- tensor::traced
- tensor_ops::abs
- tensor_ops::add
- tensor_ops::add_scalar
- tensor_ops::backward
- tensor_ops::clamp
- tensor_ops::cos
- tensor_ops::div
- tensor_ops::div_scalar
- tensor_ops::dropout
- tensor_ops::exp
- tensor_ops::ln
- tensor_ops::log_softmax
- tensor_ops::logsumexp
- tensor_ops::matmul
- tensor_ops::matmul_transpose
- tensor_ops::max
- tensor_ops::maximum
- tensor_ops::mean
- tensor_ops::min
- tensor_ops::minimum
- tensor_ops::mul
- tensor_ops::mul_scalar
- tensor_ops::nans_to
- tensor_ops::negate
- tensor_ops::normalize
- tensor_ops::powf
- tensor_ops::powi
- tensor_ops::relu
- tensor_ops::sigmoid
- tensor_ops::sin
- tensor_ops::softmax
- tensor_ops::sqrt
- tensor_ops::square
- tensor_ops::stddev
- tensor_ops::sub
- tensor_ops::sub_scalar
- tensor_ops::sum
- tensor_ops::tanh
- tensor_ops::value_mask
- tensor_ops::var
- tensor_ops::vecmat_mul
- tensor_ops::vecmat_mul_transpose