burn-tensor 0.1.0

This library provides multiple tensor implementations hidden behind an easy to use API that supports reverse mode automatic differentiation. 
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
26
27
28
29

use crate::tensor::{backend::ndarray::NdArrayTensor, ops::*, Shape};
use ndarray::{Dim, Dimension};

macro_rules! define_impl {
    (
        $n:expr
    ) => {
        impl<P, const D1: usize> TensorOpsReshape<P, D1, $n, NdArrayTensor<P, $n>>
            for NdArrayTensor<P, D1>
        where
            P: Clone + Default + std::fmt::Debug,
            Dim<[usize; $n]>: Dimension,
        {
            fn reshape(&self, shape: Shape<$n>) -> NdArrayTensor<P, $n> {
                let dim: Dim<[usize; $n]> = shape.clone().into();
                let array = self.array.reshape(dim).into_dyn();

                NdArrayTensor { array, shape }
            }
        }
    };
}

define_impl!(1);
define_impl!(2);
define_impl!(3);
define_impl!(4);
define_impl!(5);
define_impl!(6);