Merge pull request #323 from AndreiKingsley/andrew

fixes
This commit is contained in:
Iaroslav Postovalov 2021-05-08 03:34:57 +07:00 committed by GitHub
commit 512ed1139c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
6 changed files with 8 additions and 42 deletions

View File

@ -9,17 +9,17 @@ import space.kscience.kmath.tensors.api.Tensor
import space.kscience.kmath.tensors.core.internal.tensor import space.kscience.kmath.tensors.core.internal.tensor
/** /**
* Casts [Tensor<Double>] to [DoubleTensor] * Casts [Tensor] of [Double] to [DoubleTensor]
*/ */
public fun Tensor<Double>.toDoubleTensor(): DoubleTensor = this.tensor public fun Tensor<Double>.toDoubleTensor(): DoubleTensor = this.tensor
/** /**
* Casts [Tensor<Int>] to [IntTensor] * Casts [Tensor] of [Int] to [IntTensor]
*/ */
public fun Tensor<Int>.toIntTensor(): IntTensor = this.tensor public fun Tensor<Int>.toIntTensor(): IntTensor = this.tensor
/** /**
* @return [DoubleArray] of tensor elements * Returns [DoubleArray] of tensor elements
*/ */
public fun DoubleTensor.toDoubleArray(): DoubleArray { public fun DoubleTensor.toDoubleArray(): DoubleArray {
return DoubleArray(numElements) { i -> return DoubleArray(numElements) { i ->
@ -28,42 +28,10 @@ public fun DoubleTensor.toDoubleArray(): DoubleArray {
} }
/** /**
* @return [IntArray] of tensor elements * Returns [IntArray] of tensor elements
*/ */
public fun IntTensor.toIntArray(): IntArray { public fun IntTensor.toIntArray(): IntArray {
return IntArray(numElements) { i -> return IntArray(numElements) { i ->
mutableBuffer[bufferStart + i] mutableBuffer[bufferStart + i]
} }
} }
/**
* Casts [Array<DoubleArray>] to [DoubleTensor]
*/
public fun Array<DoubleArray>.toDoubleTensor(): DoubleTensor {
val n = size
check(n > 0) { "An empty array cannot be casted to tensor" }
val m = first().size
check(m > 0) { "Inner arrays must have at least 1 argument" }
check(all { size == m }) { "Inner arrays must be the same size" }
val shape = intArrayOf(n, m)
val buffer = this.flatMap { arr -> arr.map { it } }.toDoubleArray()
return DoubleTensor(shape, buffer, 0)
}
/**
* Casts [Array<IntArray>] to [IntTensor]
*/
public fun Array<IntArray>.toIntTensor(): IntTensor {
val n = size
check(n > 0) { "An empty array cannot be casted to tensor" }
val m = first().size
check(m > 0) { "Inner arrays must have at least 1 argument" }
check(all { size == m }) { "Inner arrays must be the same size" }
val shape = intArrayOf(n, m)
val buffer = this.flatMap { arr -> arr.map { it } }.toIntArray()
return IntTensor(shape, buffer, 0)
}

View File

@ -194,5 +194,3 @@ private fun DoubleTensorAlgebra.testSVDFor(tensor: DoubleTensor, epsilon: Double
assertTrue(tensor.eq(tensorSVD, epsilon)) assertTrue(tensor.eq(tensorSVD, epsilon))
} }