burn_tensor/tensor/
mod.rs

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