Function extract_bias_dim 

pub fn extract_bias_dim<A, D>(layout: impl AsRef<LayoutRef<A, D>>) -> D::Smaller
where
    D: RemoveAxis,

Extract a suitable dimension for a bias tensor from the given reference to the layout of the weight tensor.