Skip to main content

concat

Function concat 

Source
pub fn concat() -> impl Fn(Vec<SymTensor>) -> SymTensor
Expand description

Channel-wise concatenation of N NCHW tensors.

All inputs must share the same B, H, W. Channels are concatenated in order. Output shape: [B, C0+C1+...+CN, H, W].