fix NDArray cast

2021-10-20 17:04:00 +03:00 · 2021-10-20 17:04:00 +03:00 · 47aeb36979
commit 47aeb36979
parent cfd3f3b7e1
1 changed files with 2 additions and 2 deletions
--- a/kmath-multik/src/main/kotlin/space/kscience/kmath/multik/MultikTensorAlgebra.kt
+++ b/kmath-multik/src/main/kotlin/space/kscience/kmath/multik/MultikTensorAlgebra.kt
@ -171,7 +171,7 @@ public class MultikTensorAlgebra<T : Number> internal constructor(

        val mt = asMultik().array
        return if (mt.shape.contentEquals(shape)) {
-            (this as MultikTensor<T>).array
+            mt
        } else {
            NDArray(mt.data, mt.offset, shape, dim = DN(shape.size), base = mt.base ?: mt)
        }.wrap()