From 5da8670c2715315a9b86a1e4168699e6f6d81693 Mon Sep 17 00:00:00 2001
From: Christoph Groth <christoph.groth@cea.fr>
Date: Thu, 6 Mar 2025 11:40:22 +0100
Subject: Use from_fn

---
 src/main.rs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'src')

diff --git a/src/main.rs b/src/main.rs
index 796d0e4..e8f7740 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,4 +1,4 @@
-use mdarray::{view, array, tensor, Slice, Dim, Const, expr::Expression};
+use mdarray::{view, tensor, Slice, Dim, Const, expr::Expression};
 use mdarray as md;
 
 // Indexing convention: C_ij <- A_ik * B_kj
@@ -17,7 +17,9 @@ fn matmul<D0: Dim, D1: Dim, D2: Dim>(
 
 fn main() {
     let a = view![[1.0, 4.0], [2.0, 5.0], [3.0, 6.0]];
-    let b = array![[0.0, 1.0], [1.0, 1.0]];
+    let b = md::Array::<f64, (Const::<2>, Const::<2>)>::from_fn(
+        |i| if i[0] == 0 && i[1] == 0 { 0.0 } else { 1.0 }
+    );
     let b = b.reshape((Const::<2>, !0));
 
     // .into_dyn() replaces .reshape(DynRank::from_dims(&[2, 3])).into():
-- 
cgit v1.2.3-74-g4815