pub fn tensor_to_array_2<B: Backend>(x: Tensor<B, 1>) -> [usize; 2]