pub fn concatenate_device( arrays: &[impl AsRef<Array>], stream: impl AsRef<Stream>, ) -> Result<Array>
Concatenate the arrays along the first axis. Returns an error if the shapes are invalid.