List of all items

Structs

• backend::Device
• nn::BCELoss
• nn::BCEWithLogitsLoss
• nn::CrossEntropyLoss
• nn::Dropout
• nn::Linear
• nn::Lstm
• nn::MSELOSS
• nn::conv::Conv2d
• nn::conv::Conv2dConfig
• nn::conv::Convolution
• nn::pool::Pooling
• tensor::CpuDevice
• tensor::CudaDevice
• tensor::GenericTensor
• tensor::MetalDevice
• tensor::TensorIterator
• utils::NoGrad

Enums

• backend::DeviceType
• tensor::Axis

Traits

• nn::loss::SumTensor
• tensor::DeviceTypeT
• tensor::MatrixDot
• tensor::TensorExt
• utils::Overloadable
• utils::RequiresGrad

Functions

• nn::loss::cross_entropy_error
• nn::loss::gradient_descent
• nn::loss::numerical_diff
• nn::loss::numerical_gradient
• nn::loss::sum_squares_error
• utils::initialize_ops_grad

Type Aliases

• tensor::Tensor

Statics

• utils::COMPUTE_GRAD