Update om.js

lutzroeder · Jul 6, 2024 · 263ee4d · 263ee4d
1 parent aa75081
commit 263ee4d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/om.js b/source/om.js
@@ -505,7 +505,7 @@ om.Utility = class {
     }
 
     static tensorType(desc) {
-        if (desc.shape && Array.isArray(desc.shape.dim)) {
+        if (desc && desc.shape && Array.isArray(desc.shape.dim)) {
             const dataType = desc && desc.dtype ? om.Utility.dtype(desc.dtype) : '?';
             const shape = new om.TensorShape(desc.shape.dim);
             return new om.TensorType(dataType, shape, desc.layout);