List of all items
Structs
- devices::Cpu
- diff_fns::Abs
- diff_fns::BinaryAdd
- diff_fns::BinaryDiv
- diff_fns::BinaryMul
- diff_fns::BinarySub
- diff_fns::Cos
- diff_fns::Exp
- diff_fns::Ln
- diff_fns::ReLU
- diff_fns::Sigmoid
- diff_fns::Sin
- diff_fns::Square
- diff_fns::Tanh
- nn::Linear
- optim::adam::Adam
- optim::sgd::Sgd
- prelude::GradientTape
- prelude::Gradients
- prelude::NoTape
- prelude::PhantomTensor
- prelude::Tensor0D
- prelude::Tensor1D
- prelude::Tensor2D
- prelude::Tensor3D
- prelude::Tensor4D
- prelude::UniqueId
- prelude::WithTape
Enums
Traits
- devices::AllocateZeros
- devices::Device
- devices::FillElements
- devices::HasDevice
- devices::MapElements
- devices::ReduceElements
- devices::ReduceLastDim
- devices::ZipMapElements
- diff_fns::DiffBinaryFunction
- diff_fns::DifferentiableFunction
- nn::Module
- numpy::NumpyDtype
- numpy::NumpyShape
- numpy::WriteNumbers
- optim::optimizer::Optimizer
- prelude::CanPutTapeHolder
- prelude::CanUpdateWithGradients
- prelude::CountElements
- prelude::GradientProvider
- prelude::HasArrayData
- prelude::HasArrayType
- prelude::HasInner
- prelude::HasUniqueId
- prelude::IntoPhantom
- prelude::Randomize
- prelude::TapeHolder
- prelude::Tensor
- prelude::TensorCreator
- prelude::ZeroElements
- tensor_ops::HasAbsMethod
- tensor_ops::HasAbsRefMethod
- tensor_ops::HasCosMethod
- tensor_ops::HasCosRefMethod
- tensor_ops::HasExpMethod
- tensor_ops::HasExpRefMethod
- tensor_ops::HasLnMethod
- tensor_ops::HasLnRefMethod
- tensor_ops::HasMeanMethod
- tensor_ops::HasReLUMethod
- tensor_ops::HasReLURefMethod
- tensor_ops::HasSigmoidMethod
- tensor_ops::HasSigmoidRefMethod
- tensor_ops::HasSinMethod
- tensor_ops::HasSinRefMethod
- tensor_ops::HasSquareMethod
- tensor_ops::HasSquareRefMethod
- tensor_ops::HasTanhMethod
- tensor_ops::HasTanhRefMethod
Functions
- losses::cross_entropy_with_logits_loss
- losses::mae_loss
- losses::mse_loss
- numpy::save
- prelude::backward
- tensor_ops::add
- tensor_ops::add_lhs
- tensor_ops::apply
- tensor_ops::apply_binary
- tensor_ops::apply_binary_lhs
- tensor_ops::apply_ref
- tensor_ops::broadcast_inner_sub
- tensor_ops::broadcast_outer_add
- tensor_ops::clamp
- tensor_ops::div
- tensor_ops::div_lhs
- tensor_ops::log_softmax
- tensor_ops::logsumexp
- tensor_ops::matmul
- tensor_ops::mul
- tensor_ops::mul_lhs
- tensor_ops::nans_to
- tensor_ops::negate
- tensor_ops::softmax
- tensor_ops::sub
- tensor_ops::sub_lhs
- tensor_ops::sum_last_dim
- tensor_ops::value_mask
- tensor_ops::vecmat_mul