[onnx]update onnx_tensor_is_scalar function

NSKernel · Dec 1, 2020 · 97ab1d3 · 97ab1d3
1 parent c0a0604
commit 97ab1d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/onnx.h b/src/onnx.h
@@ -270,7 +270,7 @@ void onnx_tensor_apply(struct onnx_tensor_t * t, void * buf, size_t len);
 
 static inline int onnx_tensor_is_scalar(struct onnx_tensor_t * t)
 {
-	return (t->ndim > 0) ? 0 : 1;
+	return ((t->ndim == 0) && (t->ndata == 1)) ? 1 : 0;
 }
 
 static inline int onnx_tensor_broadcast_is_valid(struct onnx_tensor_t * x, int * dims, int ndim)