docs

Author: crutcher

crates/burn-tensor/src/tensor/api/numeric.rs

@@ -606,7 +606,7 @@ where
     ///         [[1.0, -2.0, 3.0], [5.0, 9.0, 6.0]],
     ///         [[9.0, 2.0, 5.0], [5.0, 7.0, 7.0]],
     ///     ], &device);
-    ///     let tensor = tensor.clone().sum_dims_squeeze(&[0, 1]);
+    ///     let tensor = tensor.clone().sum_dims_squeeze::<1, _>(&[0, 1]);
     ///     println!("{tensor}");
     ///     // [20.0, 16.0, 21.0]
     /// }