kaze/orx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[orx-tensorflow] loosen up

Browse Source
This commit is contained in:
Edwin Jakobs
2020-10-24 11:07:53 +02:00
parent b41ba5fdfc
commit 98e12d1d53
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
orx-tensorflow/src/main/kotlin/Tensor.kt
View File

@@ -36,12 +36,12 @@ fun ColorBuffer.copyTo(tensor: Tensor<TUint8>) {
fun Tensor<TFloat32>.copyTo(colorBuffer: ColorBuffer) { fun Tensor<TFloat32>.copyTo(colorBuffer: ColorBuffer) {
val s = shape() val s = shape()
require(s.numDimensions() == 2 || s.numDimensions() == 3)
val components = when { val components = when {
s.numDimensions() == 2 -> 1
s.numDimensions() == 3 -> s.size(2).toInt() s.numDimensions() == 3 -> s.size(2).toInt()
s.numDimensions() == 4 -> s.size(3).toInt() s.numDimensions() == 4 -> s.size(3).toInt()
else -> 1 else -> error("can't copy to colorbuffer from ${s.numDimensions()}D tensor")
} }
val format = when (components) { val format = when (components) {