Crate burn_tensor

Modules

• activation
• backend
• container
• loss
• module
• ops

Macros

• check
  We use a macro for all checks, since the panic message file and line number will match the function that does the check instead of a the generic error.rs crate private unreleated file and line number.
• make_element

Structs

• Bool
• Data
• DataSerialize
• DistributionSampler
• Float
• Int
• Shape
• Tensor
• bf16
  A 16-bit floating point type implementing the bfloat16 format.
• f16
  A 16-bit floating point type implementing the IEEE 754-2008 standard binary16 a.k.a half format.

Enums

• Distribution
• DistributionSamplerKind
• Precision

Traits

• BasicOps
  Trait that list all operations that can be applied on all tensors.
• Element
• ElementConversion
• ElementPrecision
• ElementRandom
• Numeric
  Trait that list all operations that can be applied on all numerical tensors.
• TensorKind