有没有一种简单的方法可以遍历多维数组并转换Swift中的每个元素？

func traverseAndTransform<T>(_ array: [[T]], transform: (T) -> T) -> [[T]] {
    var result = [[T]]()
    
    for subArray in array {
        var transformedArray = [T]()
        
        for element in subArray {
            let transformedElement = transform(element)
            transformedArray.append(transformedElement)
        }
        
        result.append(transformedArray)
    }
    
    return result
}

let inputArray = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
let transformedArray = traverseAndTransform(inputArray) { element in
    return element * 2
}

print(transformedArray)  // 输出 [[2, 4, 6], [8, 10, 12], [14, 16, 18]]