Demonstrate rank >6

1 files changed, 10 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index 3ba7bf2..796d0e4 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,4 +1,5 @@
 use mdarray::{view, array, tensor, Slice, Dim, Const, expr::Expression};
+use mdarray as md;
 
 // Indexing convention: C_ij <- A_ik * B_kj
 fn matmul<D0: Dim, D1: Dim, D2: Dim>(
@@ -39,6 +40,14 @@ fn main() {
     dbg!(std::any::type_name_of_val(&e));
 
     // permute
-    let f = c.permute((1, 0));
+    let f = c.permute([1, 0]);
     dbg!(std::any::type_name_of_val(&f));
+
+    // Arrays with rank > 6:
+    let x: Vec<f64> = (0..128).map(|x| x as f64).collect();
+    let x: md::DTensor<f64, 1> = x.into();
+    let x = x.into_dyn();
+    let x = x.reshape(&[2; 7]);
+    let x = x.permute(&[6, 5, 4, 3, 2, 1, 0]);
+    dbg!(std::any::type_name_of_val(&x));
 }