github/ezkl
1
1
Fork 0
mirror of https://github.com/zkonduit/ezkl.git synced 2026-01-10 06:48:01 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix: empty dim len can be 1 (#949)

Browse Source
This commit is contained in:
dante
2025-02-28 23:56:19 -05:00
committed by GitHub
parent bc26691bd5
commit fcbb27677f
3 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/onnx/hierarchical_risk/input.json Normal file
View File

File diff suppressed because one or more lines are too long

BIN
examples/onnx/hierarchical_risk/network.onnx Normal file
View File

Binary file not shown.

4
src/tensor/val.rs
View File

@@ -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
}