pub fn matrix_get_col( tensor: &Tensor, col: usize, ) -> Result<Tensor, RuntimeError>
Get a column from a tensor