Module prelude 

Re-exports

pub use crate::manifolds::Euclidean;

pub use crate::manifolds::Manifold;

pub use crate::optimisers::ManifoldRGD;

pub use crate::optimisers::RiemannianAdam;

pub use crate::optimisers::RiemannianAdamConfig;

Modules

nn

Neural network module.

Macros

s

Creates a slice specification for tensor indexing operations.

Structs

Bool

A type-level representation of the kind of a bool tensor.

Float

A type-level representation of the kind of a float tensor

Int

A type-level representation of the kind of a int tensor.

Shape

Shape of a tensor.

Tensor

A tensor with a given backend, shape and data type.

TensorData

Data structure for tensors.

Traits

Backend

This trait defines all types and functions needed for a backend to be used with burn.

Config

Configuration trait.

ElementConversion

Element conversion trait for tensor.

Module

Trait for all neural network modules.

Type Aliases

Device

Device type used by the backend.

Derive Macros

Config

Derive macro for the config.

Module

Derive macro for the module.