pub fn extract_bias_dim<A, D>(
layout: impl AsRef<LayoutRef<A, D>>,
) -> <D as Dimension>::Smallerwhere
D: RemoveAxis,Expand description
Extract a suitable dimension for a bias tensor from the given reference to the layout of the weight tensor.