burn_tensor/tensor/
mod.rs

1pub(crate) mod stats;
2
3mod api;
4
5pub use api::*;
6
7// Re-exported types
8pub use burn_std::{
9    BoolDType, BoolStore, DType, DataError, FloatDType, IndexingUpdateOp, IntDType, TensorData,
10    Tolerance, distribution::*, element::*, indexing::*, s, shape::*, slice::*,
11};
12
13/// The tensor kind module.
14pub mod kind;
15pub use kind::{Bool, Float, Int};
16
17/// The activation module.
18pub mod activation;
19
20/// The container module.
21pub mod container {
22    pub use burn_std::tensor::container::TensorContainer;
23}
24
25/// The grid module.
26pub mod grid;
27
28/// The linalg module.
29pub mod linalg;
30
31/// The loss module.
32pub mod loss;
33
34/// The neural network module.
35pub mod module;
36
37/// The signal processing module.
38pub mod signal;
39
40/// Operations on tensors module.
41pub mod ops {
42    pub(crate) use crate::bridge::*;
43    pub use burn_std::ops::*;
44}
45
46/// Tensor quantization module.
47pub mod quantization;
48
49#[cfg(feature = "std")]
50pub mod distributed;
51
52#[cfg(feature = "std")]
53pub use report::*;
54
55#[cfg(feature = "std")]
56mod report;
burn_tensor/tensor/mod.rs

burn_tensor/tensor/
mod.rs
