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

Co-authored-by: Corey Lowman <clowman1993@gmail.com>
coreylowman · Dec 2, 2023 · b629f64 · b629f64
1 parent 6d8ab56
commit b629f64
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 {