将矩阵声明为第32页(np.数组())

Question

 matrix = np.float32(np.array([[0.0 for i in range(dimension)] for j in range(dimension)]))

如果我想做单精度的矩阵运算，那么上面声明的数组就足够了吗，还是我必须按下面的方法对每个算术运算进行截断？你知道吗

np.float32(matrix[a][b] op matrix[c][d])

Answer 1

不，您可以指定数组的dtype：

np.array([[0.0 for i in range(dimension)] for j in range(dimension)],
         dtype=np.float32)

请注意，如果使用零，还可以使用：

np.zeros((dimension, dimension), dtype=np.float32)

通过指定dtype，数组的所有元素都具有相同的类型。您还可以指定数据的类型，例如，如果每个列的类型相同，但列可能不同，如this answer中所述。你知道吗