pub fn print_loop<T: TensorTrait<T>>( max_indices: Vec<i32>, f: fn(&Tensor<T>, Vec<i32>), t: &Tensor<T>, )