在Java 8中如何返回二维数组？

import java.util.Arrays;

public class Main {
    public static void main(String[] args) {
        int[][] matrix = generateMatrix(3, 3);
        printMatrix(matrix);
    }

    public static int[][] generateMatrix(int rows, int cols) {
        return Arrays.stream(new int[rows][cols])
                .map(row -> Arrays.stream(row)
                        .map(col -> generateValue())
                        .toArray())
                .toArray(int[][]::new);
    }

    public static int generateValue() {
        // 生成二维数组元素的逻辑
        return 0;
    }

    public static void printMatrix(int[][] matrix) {
        for (int[] row : matrix) {
            System.out.println(Arrays.toString(row));
        }
    }
}