List of all items
Structs
- Assert
- data::SubsetIterator
- devices::Cpu
- gradients::GradientTape
- gradients::Gradients
- gradients::NoneTape
- gradients::OwnedTape
- gradients::UnusedTensors
- nn::Abs
- nn::Conv2D
- nn::Cos
- nn::Dropout
- nn::DropoutOneIn
- nn::Exp
- nn::FlattenImage
- nn::GeneralizedResidual
- nn::LayerNorm1D
- nn::Linear
- nn::Ln
- nn::MultiHeadAttention
- nn::ReLU
- nn::Repeated
- nn::Residual
- nn::Sigmoid
- nn::Sin
- nn::Softmax
- nn::SplitInto
- nn::Sqrt
- nn::Square
- nn::Tanh
- nn::TransformerDecoder
- nn::TransformerDecoderBlock
- 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::HasArrayType
- arrays::HasAxis
- arrays::ZeroElements
- devices::AllocateZeros
- devices::Device
- devices::FillElements
- devices::ForEachBroadcast1
- devices::ForEachBroadcast2
- devices::ForEachBroadcast3
- devices::ForEachBroadcast4
- devices::ForEachElement
- devices::HasDevice
- devices::MatMul
- devices::MatMulOp
- devices::Reduce1Axis
- devices::ReduceAllElements
- devices::SelectAlongAxis
- devices::Transpose
- gradients::CanUpdateWithGradients
- gradients::GradientProvider
- gradients::Tape
- nn::LoadFromNpz
- nn::Module
- nn::ResetParams
- nn::SaveToNpz
- numpy::NumpyDtype
- numpy::NumpyShape
- numpy::ReadNumbers
- numpy::WriteNumbers
- optim::Optimizer
- tensor::HasArrayData
- tensor::IntoPhantom
- tensor::PutTape
- tensor::Randomize
- tensor::Tensor
- tensor::TensorCreator
- tensor_ops::Broadcast1
- tensor_ops::Broadcast2
- tensor_ops::Broadcast3
- tensor_ops::Broadcast4
- tensor_ops::MatMulTrTyping
- tensor_ops::MatMulTyping
- tensor_ops::Reduce1
- tensor_ops::Reshape
- tensor_ops::Select1
- 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::trace
- tensor::traced
- tensor_ops::abs
- tensor_ops::add_scalar
- tensor_ops::backward
- tensor_ops::binary_map::add
- tensor_ops::binary_map::div
- tensor_ops::binary_map::maximum
- tensor_ops::binary_map::minimum
- tensor_ops::binary_map::mul
- tensor_ops::binary_map::sub
- tensor_ops::clamp
- tensor_ops::conv2d
- tensor_ops::conv2d_batched
- tensor_ops::cos
- tensor_ops::div_scalar
- tensor_ops::dropout
- tensor_ops::exp
- tensor_ops::ln
- tensor_ops::log_softmax
- tensor_ops::logsumexp
- tensor_ops::map
- tensor_ops::map_df_uses_fx
- tensor_ops::matmul
- tensor_ops::matmul_transpose
- tensor_ops::max_axis
- tensor_ops::mean
- tensor_ops::mean_axis
- tensor_ops::min_axis
- tensor_ops::mul_scalar
- tensor_ops::nans_to
- tensor_ops::negate
- tensor_ops::normalize_axis
- tensor_ops::relu
- tensor_ops::sigmoid
- tensor_ops::sin
- tensor_ops::softmax
- tensor_ops::sqrt
- tensor_ops::square
- tensor_ops::std_axis
- tensor_ops::sub_scalar
- tensor_ops::sum
- tensor_ops::sum_axis
- tensor_ops::tanh
- tensor_ops::value_mask
- tensor_ops::var_axis
- tensor_ops::vecmat_mul
- tensor_ops::vecmat_mul_transpose