Struct ConcatDescriptor 

pub struct ConcatDescriptor<const N: usize> {
    pub a_shape: [i32; N],
    pub b_shape: [i32; N],
    pub concat_dim: u8,
    pub element: ElementKind,
}

Descriptor for a 2-input concat op.

a_shape and b_shape must match on every axis except concat_dim. Output shape is a_shape with [concat_dim] set to a_shape[concat_dim] + b_shape[concat_dim].

Fields

a_shape: [i32; N]

Shape of the first input.

b_shape: [i32; N]

Shape of the second input.

concat_dim: u8

Axis to concatenate along. Must satisfy 0 <= concat_dim < N.

element: ElementKind

Element type.

Implementations

impl<const N: usize> ConcatDescriptor<N>

pub fn output_shape(&self) -> [i32; N]

Compute the output shape (a_shape with [concat_dim] summed).

Trait Implementations

impl<const N: usize> Clone for ConcatDescriptor<N>

fn clone(&self) -> ConcatDescriptor<N>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<const N: usize> Copy for ConcatDescriptor<N>

impl<const N: usize> Debug for ConcatDescriptor<N>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.