complete lu and matrix mapping

This commit is contained in:
Andrei Kislitsyn 2021-03-17 17:53:14 +03:00
parent efb23591a9
commit 1fa0da2810
3 changed files with 85 additions and 95 deletions

View File

@ -1,5 +1,7 @@
package space.kscience.kmath.tensors package space.kscience.kmath.tensors
import space.kscience.kmath.linear.Matrix
import space.kscience.kmath.nd.*
import space.kscience.kmath.structures.* import space.kscience.kmath.structures.*
@ -29,59 +31,50 @@ public open class BufferedTensor<T>(
override fun hashCode(): Int = 0 override fun hashCode(): Int = 0
} // todo rename to vector
public inline fun forEachVector(vectorAction : (MutableStructure1D<T>) -> Unit): Unit {
/* check(shape.size >= 1) {"todo"}
//todo make generator mb nextMatrixIndex? val vectorOffset = strides.strides[0]
public class InnerMatrix<T>(private val tensor: BufferedTensor<T>){ val vectorShape = intArrayOf(shape.last())
private var offset: Int = 0 for (offset in 0 until numel step vectorOffset) {
private val n : Int = tensor.shape.size val vector = BufferedTensor<T>(vectorShape, buffer, offset).as1D()
//stride? vectorAction(vector)
private val step = tensor.shape[n - 1] * tensor.shape[n - 2]
public operator fun get(i: Int, j: Int): T {
val index = tensor.strides.index(offset)
index[n - 2] = i
index[n - 1] = j
return tensor[index]
}
public operator fun set(i: Int, j: Int, value: T): Unit {
val index = tensor.strides.index(offset)
index[n - 2] = i
index[n - 1] = j
tensor[index] = value
}
public fun makeStep(){
offset += step
} }
} }
public class InnerVector<T>(private val tensor: BufferedTensor<T>){ public inline fun forEachMatrix(matrixAction : (MutableStructure2D<T>) -> Unit): Unit {
private var offset: Int = 0 check(shape.size >= 2) {"todo"}
private val n : Int = tensor.shape.size val matrixOffset = strides.strides[1]
//stride? val matrixShape = intArrayOf(shape[shape.size - 2], shape.last()) //todo better way?
private val step = tensor.shape[n - 1] for (offset in 0 until numel step matrixOffset) {
val matrix = BufferedTensor<T>(matrixShape, buffer, offset).as2D()
matrixAction(matrix)
}
}
// todo remove code copy-pasting
public operator fun get(i: Int): T { public fun vectorSequence(): Sequence<MutableStructure1D<T>> = sequence {
val index = tensor.strides.index(offset) check(shape.size >= 1) {"todo"}
index[n - 1] = i val vectorOffset = strides.strides[0]
return tensor[index] val vectorShape = intArrayOf(shape.last())
for (offset in 0 until numel step vectorOffset) {
val vector = BufferedTensor<T>(vectorShape, buffer, offset).as1D()
yield(vector)
}
} }
public operator fun set(i: Int, value: T): Unit { public fun matrixSequence(): Sequence<MutableStructure2D<T>> = sequence {
val index = tensor.strides.index(offset) check(shape.size >= 2) {"todo"}
index[n - 1] = i val matrixOffset = strides.strides[1]
tensor[index] = value val matrixShape = intArrayOf(shape[shape.size - 2], shape.last()) //todo better way?
for (offset in 0 until numel step matrixOffset) {
val matrix = BufferedTensor<T>(matrixShape, buffer, offset).as2D()
yield(matrix)
}
} }
public fun makeStep(){
offset += step
} }
}
//todo default buffer = arrayOf(0)???
*/
public class IntTensor( public class IntTensor(
shape: IntArray, shape: IntArray,

View File

@ -9,27 +9,21 @@ public class DoubleLinearOpsTensorAlgebra :
} }
override fun DoubleTensor.lu(): Pair<DoubleTensor, IntTensor> { override fun DoubleTensor.lu(): Pair<DoubleTensor, IntTensor> {
/*
// todo checks // todo checks
val luTensor = this.copy() val luTensor = this.copy()
val lu = InnerMatrix(luTensor)
//stride TODO!!! move to generator?
var matCnt = 1
for (i in 0 until this.shape.size - 2) {
matCnt *= this.shape[i]
}
val n = this.shape.size val n = this.shape.size
val m = this.shape[n - 1] val m = this.shape.last()
val pivotsShape = IntArray(n - 1) { i -> val pivotsShape = IntArray(n - 1) { i -> this.shape[i] }
this.shape[i]
}
val pivotsTensor = IntTensor( val pivotsTensor = IntTensor(
pivotsShape, pivotsShape,
IntArray(matCnt * m) { 0 } IntArray(pivotsShape.reduce(Int::times)) { 0 } //todo default???
) )
val pivot = InnerVector(pivotsTensor)
for (i in 0 until matCnt) { for ((lu, pivots) in luTensor.matrixSequence().zip(pivotsTensor.vectorSequence())){
for (row in 0 until m) pivot[row] = row for (row in 0 until m) pivots[row] = row
for (i in 0 until m) { for (i in 0 until m) {
var maxA = -1.0 var maxA = -1.0
@ -47,9 +41,9 @@ public class DoubleLinearOpsTensorAlgebra :
if (iMax != i) { if (iMax != i) {
val j = pivot[i] val j = pivots[i]
pivot[i] = pivot[iMax] pivots[i] = pivots[iMax]
pivot[iMax] = j pivots[iMax] = j
for (k in 0 until m) { for (k in 0 until m) {
val tmp = lu[i, k] val tmp = lu[i, k]
@ -66,26 +60,28 @@ public class DoubleLinearOpsTensorAlgebra :
} }
} }
} }
lu.makeStep()
pivot.makeStep()
} }
return Pair(luTensor, pivotsTensor)*/
TODO("Andrei, use view, get, as2D, as1D") return Pair(luTensor, pivotsTensor)
} }
override fun luPivot(lu: DoubleTensor, pivots: IntTensor): Triple<DoubleTensor, DoubleTensor, DoubleTensor> { override fun luPivot(luTensor: DoubleTensor, pivotsTensor: IntTensor): Triple<DoubleTensor, DoubleTensor, DoubleTensor> {
/*
//todo checks //todo checks
val n = lu.shape[0] val n = luTensor.shape.last()
val p = lu.zeroesLike() val pTensor = luTensor.zeroesLike()
pivots.buffer.unsafeToIntArray().forEachIndexed { i, pivot -> for ((p, pivot) in pTensor.matrixSequence().zip(pivotsTensor.vectorSequence())){
p[i, pivot] = 1.0 for (i in 0 until n){
p[i, pivot[i]] = 1.0
}
} }
val l = lu.zeroesLike()
val u = lu.zeroesLike()
val lTensor = luTensor.zeroesLike()
val uTensor = luTensor.zeroesLike()
for ((pairLU, lu) in lTensor.matrixSequence().zip(uTensor.matrixSequence()).zip(luTensor.matrixSequence())){
val (l, u) = pairLU
for (i in 0 until n) { for (i in 0 until n) {
for (j in 0 until n) { for (j in 0 until n) {
if (i == j) { if (i == j) {
@ -99,9 +95,10 @@ public class DoubleLinearOpsTensorAlgebra :
} }
} }
} }
}
return Triple(pTensor, lTensor, uTensor)
return Triple(p, l, u)*/
TODO("Andrei, first we need implement get(Int)")
} }
override fun DoubleTensor.cholesky(): DoubleTensor { override fun DoubleTensor.cholesky(): DoubleTensor {

View File

@ -5,7 +5,7 @@ public interface TensorAlgebra<T, TensorType : TensorStructure<T>> {
public fun zeros(shape: IntArray): TensorType public fun zeros(shape: IntArray): TensorType
public fun TensorType.zeroesLike(): TensorType public fun TensorType.zeroesLike(): TensorType // mb it shouldn't be tensor but algebra method (like in numpy/torch) ?
public fun ones(shape: IntArray): TensorType public fun ones(shape: IntArray): TensorType
public fun TensorType.onesLike(): TensorType public fun TensorType.onesLike(): TensorType