Module burn::prelude

Expand description

Structs and macros used by most projects. Add use burn::prelude::* to your code to quickly get started with Burn.

Structs§

  • A type-level representation of the kind of a bool tensor.
  • A type-level representation of the kind of a float tensor
  • A type-level representation of the kind of a int tensor.
  • Shape of a tensor.
  • A tensor with a given backend, shape and data type.
  • Data structure for tensors.

Traits§

  • This trait defines all types and functions needed for a backend to be used with burn.
  • Configuration trait.
  • Element conversion trait for tensor.
  • Trait for all neural network modules.

Type Aliases§

  • Device type used by the backend.

Derive Macros§

  • Derive macro for the config.
  • Derive macro for the module.