From 98e12d1d5387224ad643ee6a905811b2824f3488 Mon Sep 17 00:00:00 2001
From: Edwin Jakobs <edwin@rndr.studio>
Date: Sat, 24 Oct 2020 11:07:53 +0200
Subject: [PATCH] [orx-tensorflow] loosen up

---
 orx-tensorflow/src/main/kotlin/Tensor.kt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/orx-tensorflow/src/main/kotlin/Tensor.kt b/orx-tensorflow/src/main/kotlin/Tensor.kt
index bc55525b..6cfa163a 100644
--- a/orx-tensorflow/src/main/kotlin/Tensor.kt
+++ b/orx-tensorflow/src/main/kotlin/Tensor.kt
@@ -36,12 +36,12 @@ fun ColorBuffer.copyTo(tensor: Tensor<TUint8>) {
 
 fun Tensor<TFloat32>.copyTo(colorBuffer: ColorBuffer) {
     val s = shape()
-    require(s.numDimensions() == 2 || s.numDimensions() == 3)
 
     val components = when {
+        s.numDimensions() == 2 -> 1
         s.numDimensions() == 3 -> s.size(2).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) {