This commit is contained in:
Andrei Kislitsyn 2021-03-30 15:13:45 +03:00
parent 2503d35ba8
commit 139534fdb3

View File

@ -183,11 +183,7 @@ public class DoubleLinearOpsTensorAlgebra :
}
private fun MutableStructure1D<Double>.l2Norm(): Double {
var squareSum = 0.0
for (i in 0 until size) {
squareSum += this[i] * this[i]
}
return sqrt(squareSum)
return sqrt((0 until size).sumOf { this[it] * this[it] })
}
fun qrHelper(
@ -223,7 +219,7 @@ public class DoubleLinearOpsTensorAlgebra :
val (q, r) = qr
qrHelper(matrix.as2D(), q.as2D(), r.as2D())
}
return Pair(qTensor, rTensor)
return qTensor to rTensor
}
override fun DoubleTensor.svd(): Triple<DoubleTensor, DoubleTensor, DoubleTensor> {