save

BrosnanYuen · Sep 6, 2023 · 063fdb5 · 063fdb5
1 parent faf451a
commit 063fdb5
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/Interpolate/Linear.rs b/src/Interpolate/Linear.rs
@@ -15,6 +15,11 @@ pub fn run<Z: arrayfire::FloatingPoint>(
 	,dfdt: &arrayfire::Array<Z>
 	,s: &arrayfire::Array<Z>)
 	-> arrayfire::Array<Z>  {
+
+	let t = arrayfire::transpose(t, false);
+	let f = arrayfire::transpose(f, false);
+	let dfdt = arrayfire::transpose(dfdt, false);
+	let s = arrayfire::transpose(s, false);
 
 	let s_dims = s.dims();
 	let s_num = s_dims[0];
@@ -64,7 +69,7 @@ pub fn run<Z: arrayfire::FloatingPoint>(
 
 	let result = f_init + arrayfire::mul(&dfdt_init, &step, false);
 
-	result
+	arrayfire::transpose(&result, false)
 }