fix: empty dim len can be 1

2026-01-13 16:27:59 -05:00 · 2025-02-28 14:17:00 -05:00
3 changed files with 4 additions and 1 deletions
--- a/examples/onnx/hierarchical_risk/input.json
+++ b/examples/onnx/hierarchical_risk/input.json
--- a/examples/onnx/hierarchical_risk/network.onnx
+++ b/examples/onnx/hierarchical_risk/network.onnx
--- a/src/tensor/val.rs
+++ b/src/tensor/val.rs
@@ -1342,9 +1342,11 @@ impl<F: PrimeField + TensorType + PartialOrd + std::hash::Hash> ValTensor<F> {
    /// Gets the total number of elements in the tensor
    pub fn len(&self) -> usize {
        match self {
-            ValTensor::Value { dims, .. } => {
+            ValTensor::Value { dims, inner, .. } => {
                if !dims.is_empty() && (dims != &[0]) {
                    dims.iter().product::<usize>()
+                } else if dims.is_empty() {
+                    inner.inner.len()
                } else {
                    0
                }