Crate burn_tensor

Expand description

This library provides multiple tensor implementations hidden behind an easy to use API that supports reverse mode automatic differentiation.

Modules

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
Macro to implement the element trait for a type.

Bool
A type-level representation of the kind of a bool tensor.
Data
Data structure for tensors.
DataSerialize
Data structure for serializing and deserializing tensor data.
DistributionSampler
Distribution sampler for random value of a 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.
T
Transpose marker (zero-size type). Used to sugar the transpose of a tensor, e.g.
Tensor
A tensor with a given backend, shape and data type.
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.

BasicOps
Trait that list all operations that can be applied on all tensors.
Element
Element trait for tensor.
ElementConversion
Element conversion trait for tensor.
ElementPrecision
Element precision trait for tensor.
ElementRandom
Element trait for random value of a tensor.
Numeric
Trait that list all operations that can be applied on all numerical tensors.
TensorKind
A type-level representation of the kind of a tensor.