kscience/kmath
9
0
Fork 1
Code Issues Pull Requests 6 Packages Projects Releases 16 Wiki Activity

Dev #194

Merged
altavir merged 266 commits from dev into master 2021-01-20 17:32:32 +03:00
Conversation 0 Commits 266 Files Changed 344 +3 -2
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 9a4dd31507 - Show all commits

3
kmath-nd4j/src/main/kotlin/scientifik.kmath.nd4j/Arrays.kt Normal file
View File

@ -0,0 +1,3 @@
package scientifik.kmath.nd4j
internal fun narrowToIntArray(la: LongArray): IntArray = IntArray(la.size) { la[it].toInt() }

2
kmath-nd4j/src/main/kotlin/scientifik.kmath.nd4j/ND4JStructure.kt
View File

@ -3,8 +3,6 @@ package scientifik.kmath.nd4j
import org.nd4j.linalg.api.ndarray.INDArray import org.nd4j.linalg.api.ndarray.INDArray
import scientifik.kmath.structures.NDStructure import scientifik.kmath.structures.NDStructure
internal fun narrowToIntArray(la: LongArray): IntArray = IntArray(la.size) { la[it].toInt() }
data class ND4JStructure<T>(val ndArray: INDArray) : NDStructure<INDArray> { data class ND4JStructure<T>(val ndArray: INDArray) : NDStructure<INDArray> {
override val shape: IntArray override val shape: IntArray
get() = narrowToIntArray(ndArray.shape()) get() = narrowToIntArray(ndArray.shape())