Update dfdx-core/src/tensor_ops/concat_shape_along/mod.rs

Co-authored-by: Corey Lowman <[email protected]>
coreylowman · Dec 4, 2023 · 7c88fb6 · 7c88fb6
1 parent 2a31bdf
commit 7c88fb6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dfdx-core/src/tensor_ops/concat_shape_along/mod.rs b/dfdx-core/src/tensor_ops/concat_shape_along/mod.rs
@@ -43,7 +43,7 @@ use crate::{shapes::*, tensor::*};
 /// assert_eq!(c, (Const::<2>, 6));
 /// ```
 pub trait TryConcatShapeAlong<Ax>: Sized {
-    type Output;
+    type Output: Shape;
 
     /// Concatenates self along the given axis.
     fn concat_shape_along(self, ax: Ax) -> Self::Output {