Trait dfdx::tensor_ops::TryConcat

source ·

pub trait TryConcat<Rhs>: HasErr {
    type Output;

    // Required method
    fn try_concat(self, rhs: Rhs) -> Result<Self::Output, Self::Err>;

    // Provided method
    fn concat(self, rhs: Rhs) -> Self::Output { ... }
}

👎Deprecated: Use TryConcatAlong instead

Expand description

Concatenate two tensors along the first dimension.

Pytorch equivalent torch.concat.

Concat with const dims requires nightly:

let a: Tensor<Rank2<3, 4>, f32, _> = dev.zeros();
let b: Tensor<Rank2<3, 4>, f32, _> = dev.zeros();
let _: Tensor<Rank2<6, 4>, f32, _> = a.concat(b);

Concat with usize dims:

let a: Tensor<(usize, Const<3>), f32, _> = dev.zeros_like(&(2, Const));
let b: Tensor<(usize, Const<3>), f32, _> = dev.zeros_like(&(4, Const));
let c: Tensor<(usize, Const<3>), f32, _> = a.concat(b);
assert_eq!(c.shape().0, 6);