List of all items
Structs
- data::SubsetIterator
- devices::Cpu
- gradients::GradientTape
- gradients::Gradients
- gradients::NoTape
- gradients::OwnsTape
- nn::Abs
- nn::Cos
- nn::Exp
- nn::Linear
- nn::Ln
- nn::ReLU
- nn::Sigmoid
- nn::Sin
- nn::Sqrt
- nn::Square
- nn::Tanh
- optim::Adam
- optim::Sgd
- tensor::PhantomTensor
- tensor::Tensor0D
- tensor::Tensor1D
- tensor::Tensor2D
- tensor::Tensor3D
- tensor::Tensor4D
- unique_id::UniqueId
Enums
Traits
- arrays::CountElements
- arrays::HasArrayType
- arrays::ZeroElements
- devices::AllocateZeros
- devices::Device
- devices::FillElements
- devices::HasDevice
- devices::MapElements
- devices::ReduceElements
- devices::ReduceLastDim
- devices::ZipMapElements
- gradients::CanUpdateWithGradients
- gradients::GradientProvider
- gradients::Tape
- nn::Module
- numpy::NumpyDtype
- numpy::NumpyShape
- numpy::WriteNumbers
- optim::Optimizer
- tensor::HasArrayData
- tensor::IntoPhantom
- tensor::PutTape
- tensor::Randomize
- tensor::Tensor
- tensor::TensorCreator
- unique_id::HasUniqueId
Functions
- data::one_hot_encode
- losses::cross_entropy_with_logits_loss
- losses::mae_loss
- losses::mse_loss
- losses::rmse_loss
- numpy::save
- tensor::backward
- tensor_ops::abs
- tensor_ops::add
- tensor_ops::binary_map
- tensor_ops::broadcast_inner_sub
- tensor_ops::broadcast_outer_add
- tensor_ops::clamp
- tensor_ops::cloned_map
- tensor_ops::cos
- tensor_ops::div
- tensor_ops::exp
- tensor_ops::ln
- tensor_ops::log_softmax
- tensor_ops::logsumexp
- tensor_ops::map
- tensor_ops::matmul
- tensor_ops::mean
- tensor_ops::mul
- tensor_ops::nans_to
- tensor_ops::negate
- tensor_ops::relu
- tensor_ops::sigmoid
- tensor_ops::sin
- tensor_ops::softmax
- tensor_ops::sqrt
- tensor_ops::square
- tensor_ops::sub
- tensor_ops::sum_last_dim
- tensor_ops::tanh
- tensor_ops::value_mask
- tensor_ops::vecmat_mul