pub trait Unsqueeze { type Output; // Required method fn unsqueeze(self, axis: usize) -> Self::Output; }