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    DType, DataError, FloatDType, IntDType, TensorData, TensorMetadata, TensorPrimitive, Tolerance,
10    distribution::*,
11    element::*,
12    indexing::*,
13    ops::TransactionPrimitive,
14    shape::*,
15    slice::*,
16    tensor::{Bool, Float, Int, TensorKind},
17};
18
19/// The activation module.
20pub mod activation;
21
22/// The backend module.
23pub mod backend {
24    pub use burn_backend::backend::*;
25}
26
27/// The container module.
28pub mod container {
29    pub use burn_backend::tensor::TensorContainer;
30}
31
32/// The grid module.
33pub mod grid;
34
35/// The linalg module.
36pub mod linalg;
37
38/// The loss module.
39pub mod loss;
40
41/// The neural network module.
42pub mod module;
43
44/// Operations on tensors module.
45pub mod ops {
46    pub use burn_backend::backend::ops::*;
47    pub use burn_backend::tensor::{
48        BoolElem, BoolTensor, Device, FloatElem, FloatTensor, IntElem, IntTensor, QuantizedTensor,
49    };
50}
51
52/// Tensor quantization module.
53pub mod quantization;
54
55#[cfg(feature = "std")]
56pub use report::*;
57
58#[cfg(feature = "std")]
59mod report;
60
61#[cfg(feature = "experimental-named-tensor")]
62mod named;
63
64#[cfg(feature = "experimental-named-tensor")]
65pub use named::*;
66
67pub 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
