arrays - Swift 中的任意矩阵或数组大小

Question

我熟悉使用 NumPy 在 Python 中创建 MxN 矩阵，例如:

In [1]: import numpy as np

In [2]: np.zeros((10,5))
Out[2]:
array([[ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.]]) 

在 Swift 中，SIMD 可以创建小型矩阵并对其执行运算(见下文)，但它似乎仅限于 4x4 矩阵的最大大小。

import simd

let mat = float2x2([[1,2],[3,4]])
mat[0]*9

有没有办法在 Swift 中定义任意大小的 MxN 维度矩阵并对该矩阵执行操作？

Answer 1

如果“矩阵”是指“数组的数组”，那么您可以使用以下语法:

var matrix = Array(count: 10, repeatedValue: Array(count: 5, repeatedValue: 0.0))

这会创建一个 10 x 5 的零矩阵。