burn-tensor 0.16.1

Tensor library with user-friendly APIs and automatic differentiation support
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13

pub(crate) mod gelu;
pub(crate) mod hard_sigmoid;
pub(crate) mod leaky_relu;
pub(crate) mod log_sigmoid;
pub(crate) mod mish;
pub(crate) mod prelu;
pub(crate) mod relu;
pub(crate) mod sigmoid;
pub(crate) mod silu;
pub(crate) mod softmax;
pub(crate) mod softmin;
pub(crate) mod softplus;
pub(crate) mod tanh_activation;