burn-tensor 0.18.0

Tensor library with user-friendly APIs and automatic differentiation support
1
2
3
4
5
6

mod base;

/// Tensor element casting.
pub mod cast;

pub use base::*;