Fix squeeze (#121)

v0lta · web-flow · commit c01682e9c9da · 2025-11-23T23:49:55.000+01:00
diff --git a/examples/network_compression/wavelet_linear.py b/examples/network_compression/wavelet_linear.py
@@ -73,7 +73,7 @@ def wavelet_analysis(self, x):
         assert shape_lst == self.coefficient_len_lst[::-1], (
             "Wavelet shape assumptions false. This is a bug."
         )
-        return c_tensor
+        return c_tensor.squeeze(1)
 
     def wavelet_reconstruction(self, x):
         """Reconstruction from a tensor input.

Original file line number	Diff line number	Diff line change
`@@ -73,7 +73,7 @@ def wavelet_analysis(self, x):`
`73`	`73`	`assert shape_lst == self.coefficient_len_lst[::-1], (`
`74`	`74`	`"Wavelet shape assumptions false. This is a bug."`
`75`	`75`	`)`
`76`		`- return c_tensor`
	`76`	`+ return c_tensor.squeeze(1)`
`77`	`77`
`78`	`78`	`def wavelet_reconstruction(self, x):`
`79`	`79`	`"""Reconstruction from a tensor input.`