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    TensorReadError, 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 loss module.
29pub mod loss;
30
31/// The neural network module.
32pub mod module;
33
34/// Operations on tensors module.
35pub mod ops {
36    pub(crate) use crate::bridge::*;
37    pub use burn_std::ops::*;
38}
39
40/// Tensor quantization module.
41pub mod quantization;
42
43#[cfg(feature = "std")]
44pub mod distributed;
45
46#[cfg(feature = "std")]
47pub use report::*;
48
49#[cfg(feature = "std")]
50mod report;
burn_tensor/tensor/mod.rs

burn_tensor/tensor/
mod.rs
