fix NDArray cast
This commit is contained in:
parent
cfd3f3b7e1
commit
47aeb36979
@ -171,7 +171,7 @@ public class MultikTensorAlgebra<T : Number> internal constructor(
|
|||||||
|
|
||||||
val mt = asMultik().array
|
val mt = asMultik().array
|
||||||
return if (mt.shape.contentEquals(shape)) {
|
return if (mt.shape.contentEquals(shape)) {
|
||||||
(this as MultikTensor<T>).array
|
mt
|
||||||
} else {
|
} else {
|
||||||
NDArray(mt.data, mt.offset, shape, dim = DN(shape.size), base = mt.base ?: mt)
|
NDArray(mt.data, mt.offset, shape, dim = DN(shape.size), base = mt.base ?: mt)
|
||||||
}.wrap()
|
}.wrap()
|
||||||
|
Loading…
Reference in New Issue
Block a user