如何从函数返回多维数组

#include <stdio.h>
#include <stdlib.h>

// 函数返回一个二维数组
int** createArray(int rows, int cols) {
    int** arr = (int**)malloc(rows * sizeof(int*));
    for (int i = 0; i < rows; i++) {
        arr[i] = (int*)malloc(cols * sizeof(int));
        for (int j = 0; j < cols; j++) {
            arr[i][j] = i * cols + j;
        }
    }
    return arr;
}

int main() {
    int rows = 3;
    int cols = 4;

    // 调用函数创建二维数组
    int** arr = createArray(rows, cols);

    // 访问二维数组的元素
    for (int i = 0; i < rows; i++) {
        for (int j = 0; j < cols; j++) {
            printf("%d ", arr[i][j]);
        }
        printf("\n");
    }

    // 释放动态分配的内存
    for (int i = 0; i < rows; i++) {
        free(arr[i]);
    }
    free(arr);

    return 0;
}