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

Refactor/ndalgebra #197

Merged
altavir merged 11 commits from refactor/ndalgebra into dev 2021-01-28 20:13:39 +03:00
Conversation 0 Commits 11 Files Changed 69 +3 -5
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 0baec14059 - Show all commits

2
kmath-core/src/commonMain/kotlin/kscience/kmath/nd/BufferNDAlgebra.kt
View File

@ -50,7 +50,7 @@ public interface BufferNDAlgebra<T, C> : NDAlgebra<T, C> {
val aBuffer = a.ndBuffer val aBuffer = a.ndBuffer
val bBuffer = b.ndBuffer val bBuffer = b.ndBuffer
val buffer = bufferFactory(strides.linearSize) { offset -> val buffer = bufferFactory(strides.linearSize) { offset ->
elementContext.transform(aBuffer.buffer[offset], bBuffer[offset]) elementContext.transform(aBuffer.buffer[offset], bBuffer.buffer[offset])
} }
return NDBuffer(strides, buffer) return NDBuffer(strides, buffer)
} }

6
kmath-core/src/commonMain/kotlin/kscience/kmath/nd/NDStructure.kt
View File

@ -217,9 +217,7 @@ public class DefaultStrides private constructor(override val shape: IntArray) :
} }
override fun offset(index: IntArray): Int = index.mapIndexed { i, value -> override fun offset(index: IntArray): Int = index.mapIndexed { i, value ->
if (value < 0 || value >= shape[i]) if (value < 0 || value >= shape[i]) throw IndexOutOfBoundsException("Index $value out of shape bounds: (0,${this.shape[i]})")
throw IndexOutOfBoundsException("Index $value out of shape bounds: (0,${this.shape[i]})")
value * strides[i] value * strides[i]
}.sum() }.sum()
@ -332,7 +330,7 @@ public inline fun <T, reified R : Any> NDStructure<T>.mapToBuffer(
public class MutableNDBuffer<T>( public class MutableNDBuffer<T>(
strides: Strides, strides: Strides,
buffer: MutableBuffer<T>, buffer: MutableBuffer<T>,
) : NDBuffer<T>(strides,buffer), MutableNDStructure<T> { ) : NDBuffer<T>(strides, buffer), MutableNDStructure<T> {
init { init {
require(strides.linearSize == buffer.size) { require(strides.linearSize == buffer.size) {