burn-tensor 0.18.0

Tensor library with user-friendly APIs and automatic differentiation support
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

pub(crate) mod check;

mod argwhere;
mod autodiff;
mod base;
mod bool;
mod cartesian_grid;
mod float;
mod int;
mod kind;
mod numeric;
mod slice;
mod sort;
mod transaction;

pub use argwhere::argwhere_data;
pub use autodiff::*;
pub use base::*;
pub use cartesian_grid::cartesian_grid;
pub use float::{DEFAULT_ATOL, DEFAULT_RTOL};
pub use kind::*;
pub use numeric::*;
pub use slice::*;
pub use sort::{argsort, sort, sort_with_indices};
pub use transaction::*;