use ndarray::RemoveAxis;
pub trait GetBiasDim<A, D>
where
D: RemoveAxis,
{
type Output;
fn get_bias_dim(&self) -> Self::Output;
}
impl<A, D, U> GetBiasDim<A, D> for U
where
D: RemoveAxis,
U: AsRef<ndarray::LayoutRef<A, D>>,
{
type Output = D::Smaller;
fn get_bias_dim(&self) -> Self::Output {
crate::extract_bias_dim(self)
}
}