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

KMP library for tensors #300

Merged
grinisrit merged 215 commits from feature/tensor-algebra into dev 2021-05-08 09:48:04 +03:00
Conversation 1 Commits 215 Files Changed 40 +7 -1
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit a1cbd7a457 - Show all commits

8
kmath-tensors/src/commonMain/kotlin/space/kscience/kmath/tensors/core/algebras/TensorLinearStructure.kt
View File

@ -56,7 +56,12 @@ internal fun stepIndex(index: IntArray, shape: IntArray, nDim: Int): IntArray {
return res return res
} }
/**
* This [Strides] implemetation follow the last dimension first convention
* For more information: https://numpy.org/doc/stable/reference/generated/numpy.ndarray.strides.html
*
* @param shape the shape of the tensor.
*/
public class TensorLinearStructure(override val shape: IntArray) : Strides public class TensorLinearStructure(override val shape: IntArray) : Strides
{ {
override val strides: IntArray override val strides: IntArray
@ -65,6 +70,7 @@ public class TensorLinearStructure(override val shape: IntArray) : Strides
override fun index(offset: Int): IntArray = override fun index(offset: Int): IntArray =
indexFromOffset(offset, strides, shape.size) indexFromOffset(offset, strides, shape.size)
// TODO: documentation (Alya)
public fun stepIndex(index: IntArray): IntArray = public fun stepIndex(index: IntArray): IntArray =
stepIndex(index, shape, shape.size) stepIndex(index, shape, shape.size)