burn_tensor/tensor/
mod.rs

1pub(crate) mod stats;
2
3mod api;
4
5pub use api::*;
6
7// Re-exported types
8pub use burn_backend::{
9    BoolDType, BoolStore, DType, DataError, FloatDType, IntDType, TensorData, TensorMetadata,
10    TensorPrimitive, Tolerance,
11    distribution::*,
12    element::*,
13    indexing::*,
14    ops::TransactionPrimitive,
15    shape::*,
16    slice::*,
17    tensor::{Bool, Float, Int, TensorKind},
18};
19
20/// The activation module.
21pub mod activation;
22
23/// The backend module.
24pub mod backend {
25    pub use burn_backend::backend::*;
26}
27
28/// The container module.
29pub mod container {
30    pub use burn_backend::tensor::TensorContainer;
31}
32
33/// The grid module.
34pub mod grid;
35
36/// The linalg module.
37pub mod linalg;
38
39/// The loss module.
40pub mod loss;
41
42/// The neural network module.
43pub mod module;
44
45/// The signal processing module.
46pub mod signal;
47
48/// Operations on tensors module.
49pub mod ops {
50    pub use burn_backend::backend::ops::*;
51    pub use burn_backend::tensor::{
52        BoolElem, BoolTensor, Device, FloatElem, FloatTensor, IntElem, IntTensor, QuantizedTensor,
53    };
54}
55
56/// Tensor quantization module.
57pub mod quantization;
58
59#[cfg(feature = "std")]
60pub use report::*;
61
62#[cfg(feature = "std")]
63mod report;
64
65pub use ops::Device; // Re-export device so that it's available from `burn_tensor::Device`.
burn_tensor/tensor/mod.rs

burn_tensor/tensor/
mod.rs
