extract_bias_dim

Function extract_bias_dim 

Source
pub fn extract_bias_dim<A, D>(
    layout: impl AsRef<LayoutRef<A, D>>,
) -> <D as Dimension>::Smaller
where D: RemoveAxis,
Expand description

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