burn_tensor/
lib.rs

1#![cfg_attr(not(feature = "std"), no_std)]
2#![warn(missing_docs)]
3#![cfg_attr(docsrs, feature(doc_cfg))]
4
5//! This library provides the core abstractions required to run tensor operations with Burn.
6//! `Tensor`s are generic over the backend to allow users to perform operations using different `Backend` implementations.
7//! Burn's tensors also support auto-differentiation thanks to the `AutodiffBackend` trait.
8
9#[macro_use]
10extern crate derive_new;
11
12extern crate alloc;
13
14mod tensor;
15
16pub(crate) use tensor::check::macros::check;
17pub use tensor::*;
18
19// Re-exported types
20pub use burn_backend::{AllocationProperty, Bytes, StreamId, bf16, f16, read_sync, try_read_sync};