Crate burn_backend 

This library provides the core types that define how Burn tensor data is represented, stored, and interpreted.

Re-exports

pub use distribution::*;

pub use element::*;

pub use backend::*;

pub use shape::*;

pub use slice::*;

pub use indexing::*;

Modules

backend

Backend trait and required types.

distribution

Random value distributions used to initialize and populate tensor data.

element

Traits and helpers for working with element types and conversions.

indexing

Indexing utilities.

quantization

Quantization data representation.

shape

Shape definition.

slice

Slice utilities.

tensor

Backend tensor primitives and operations.

Macros

dequant_op_flow

Automatically applies dequantization -> float operation [-> quantization].

dequant_op_quant

Automatically applies dequantization -> float operation -> quantization.

doc_tensor

Convenience macro to link to the burn-tensor docs for this crate version.

make_element

Macro to implement the element trait for a type.

Structs

Bytes

A buffer similar to Box<[u8]> that supports custom memory alignment and allows trailing uninitialized bytes.

StreamId

Unique identifier that can represent a stream based on the current thread id.

TensorData

Data structure for tensors.

Tolerance

The tolerance used to compare to floating point numbers.

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

AllocationProperty

The kind of allocation behind the Bytes type.

DType

DataError

The things that can go wrong when manipulating tensor data.

FloatDType

IntDType

Functions

read_sync

Read a future synchronously.

try_read_sync

Read a future synchronously.