Small fix - createUninitialized scalar utf8

deeplearning4j · Mar 12, 2019 · 3c4786f · 3c4786f
1 parent e4592dc
commit 3c4786f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/factory/Nd4j.java b/nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/factory/Nd4j.java
@@ -4880,8 +4880,12 @@ public static INDArray createUninitialized(DataType type, long... shape) {
     }
 
     public static INDArray createUninitialized(DataType type, long[] shape, char ordering) {
-        if (shape.length == 0)
+        if (shape.length == 0) {
+            if(type == DataType.UTF8){
+                return scalar("");
+            }
             return scalar(type, 0);
+        }
 
         checkShapeValues(shape);