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

mutableBuffer setter method now is used in PowerMethod

Browse Source
This commit is contained in:
margarita0303 2022-08-02 19:40:10 +03:00
parent 7e7dc9f9aa
commit 9eb953228a
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
kmath-tensors/src/commonMain/kotlin/space/kscience/kmath/tensors/core/internal/linUtils.kt
View File

@ -308,6 +308,14 @@ internal fun DoubleTensorAlgebra.svdPowerMethodHelper(
val res = ArrayList<Triple<Double, DoubleTensor, DoubleTensor>>(0) val res = ArrayList<Triple<Double, DoubleTensor, DoubleTensor>>(0)
val (matrixU, matrixS, matrixV) = USV val (matrixU, matrixS, matrixV) = USV
val matrixUStart = matrixU.bufferStart
val matrixSStart = matrixS.bufferStart
val matrixVStart = matrixV.bufferStart
val matrixUBuffer = matrixU.mutableBuffer
val matrixSBuffer = matrixS.mutableBuffer
val matrixVBuffer = matrixV.mutableBuffer
for (k in 0 until min(n, m)) { for (k in 0 until min(n, m)) {
var a = matrix.copy() var a = matrix.copy()
for ((singularValue, u, v) in res.slice(0 until k)) { for ((singularValue, u, v) in res.slice(0 until k)) {
@ -341,13 +349,13 @@ internal fun DoubleTensorAlgebra.svdPowerMethodHelper(
val uBuffer = res.map { it.second }.flatMap { it.mutableBuffer.array().toList() }.toDoubleArray() val uBuffer = res.map { it.second }.flatMap { it.mutableBuffer.array().toList() }.toDoubleArray()
val vBuffer = res.map { it.third }.flatMap { it.mutableBuffer.array().toList() }.toDoubleArray() val vBuffer = res.map { it.third }.flatMap { it.mutableBuffer.array().toList() }.toDoubleArray()
for (i in uBuffer.indices) { for (i in uBuffer.indices) {
matrixU.mutableBuffer.array()[matrixU.bufferStart + i] = uBuffer[i] matrixUBuffer[matrixUStart + i] = uBuffer[i]
} }
for (i in s.indices) { for (i in s.indices) {
matrixS.mutableBuffer.array()[matrixS.bufferStart + i] = s[i] matrixSBuffer[matrixSStart + i] = s[i]
} }
for (i in vBuffer.indices) { for (i in vBuffer.indices) {
matrixV.mutableBuffer.array()[matrixV.bufferStart + i] = vBuffer[i] matrixVBuffer[matrixVStart + i] = vBuffer[i]
} }
} }