chela 0.0.2

High-performance Machine Learning, Auto-Differentiation and Tensor Algebra crate for Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

pub mod methods;
pub mod ops;
pub mod constructors;
pub mod equals;
pub mod autograd;
pub mod print;

use crate::gradient_function::GradientFunction;
use crate::ndarray::flags::NdArrayFlags;
use crate::{NdArray, TensorDataType};

pub struct Tensor<'a, T: TensorDataType> {
    array: NdArray<'a, T>,

    pub(super) flags: NdArrayFlags,
    pub(super) grad_fn: GradientFunction<T>,
}