width, channels) while channels_first corresponds to inputs with shape (batch_size, channels, height, width).
    rF