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

use crate::tensor::Shape;
use ndarray::Dim;

macro_rules! define_convertion {
    (
        $n:expr
    ) => {
        impl Into<Dim<[usize; $n]>> for Shape<$n> {
            fn into(self) -> Dim<[usize; $n]> {
                Dim(self.dims)
            }
        }
    };
}

define_convertion!(0);
define_convertion!(1);
define_convertion!(2);
define_convertion!(3);
define_convertion!(4);
define_convertion!(5);
define_convertion!(6);