fix: handle [] shapes in sort

docs: document arguments better (#950 )
fix: empty dim len can be 1 (#949 )
2026-01-13 08:17:57 -05:00 · 2025-03-07 14:32:10 -05:00 · 2025-03-05 16:10:50 -05:00 · 2025-02-28 23:56:19 -05:00
2 changed files with 1106 additions and 122 deletions
--- a/src/circuit/ops/layouts.rs
+++ b/src/circuit/ops/layouts.rs
--- a/src/tensor/mod.rs
+++ b/src/tensor/mod.rs
@@ -926,6 +926,9 @@ impl<T: Clone + TensorType> Tensor<T> {
                ));
            }
            self.dims = vec![];
+        }
+        if self.dims() == &[0] && new_dims.iter().product::<usize>() == 1 {
+            self.dims = Vec::from(new_dims);
        } else {
            let product = if new_dims != [0] {
                new_dims.iter().product::<usize>()
@@ -1104,6 +1107,10 @@ impl<T: Clone + TensorType> Tensor<T> {
            let mut output = self.clone();
            output.reshape(shape)?;
            return Ok(output);
+        } else if self.dims() == &[0] && shape.iter().product::<usize>() == 1 {
+            let mut output = self.clone();
+            output.reshape(shape)?;
+            return Ok(output);
        }

        if self.dims().len() > shape.len() {
Author	SHA1	Message	Date
dante	fdc93b9572	fix: handle [] shapes in sort	2025-03-07 14:32:10 -05:00
dante	f631445e26	docs: document arguments better (#950 )	2025-03-05 16:10:50 -05:00
dante	fcbb27677f	fix: empty dim len can be 1 (#949 )	2025-02-28 23:56:19 -05:00