burn_tensor/tensor/
mod.rs

1pub(crate) mod stats;
2
3mod api;
4mod data;
5mod distribution;
6mod element;
7mod shape;
8
9pub use api::*;
10pub use burn_common::bytes::*;
11pub use data::*;
12pub use distribution::*;
13pub use element::*;
14pub use shape::*;
15
16/// The activation module.
17pub mod activation;
18
19/// The backend module.
20pub mod backend;
21
22/// The container module.
23pub mod container;
24
25/// The grid module.
26pub mod grid;
27
28/// The indexing module.
29pub mod indexing;
30
31pub use indexing::AsIndex;
32
33/// The linalg module.
34pub mod linalg;
35
36/// The loss module.
37pub mod loss;
38
39/// The burn module.
40pub mod module;
41
42/// Operations on tensors module.
43pub mod ops;
44
45/// Tensor quantization module.
46pub mod quantization;
47
48#[cfg(feature = "std")]
49pub use report::*;
50
51#[cfg(feature = "std")]
52mod report;
53
54#[cfg(feature = "experimental-named-tensor")]
55mod named;
56
57#[cfg(feature = "experimental-named-tensor")]
58pub use named::*;
59
60pub use ops::Device; // Re-export device so that it's available from `burn_tensor::Device`.