如果返回多维数组，如何设计此方法？

# 使用嵌套数组创建多维数组
def create_multi_array(rows, cols):
    multi_array = [[0] * cols for _ in range(rows)]
    return multi_array

# 使用数组的数组创建多维数组
def create_multi_array(rows, cols):
    multi_array = [0] * (rows * cols)
    multi_array = [multi_array[i:i + cols] for i in range(0, len(multi_array), cols)]
    return multi_array