python + assert 如何断言两个高维（包括二维）矩阵中每一个元素都相等

在上一篇博客，我实现了如何断言一个向量相等
这篇博客，其实一模一样，将一个高维矩阵变成一个向量之后再利用上次的方法就可以了。
其实，断言向量，也就是将向量变成一个个数就行了。这是不变的原理。

import numpy as np

a = np.zeros((3, 4))
b = np.zeros((3, 4))

a = a.flatten()
b = b.flatten()
assert all(a[i] == b[i] for i in range(len(a))), 'They aren\'t always equal.'

a[0] = 1
assert all(a[i] == b[i] for i in range(len(a))), 'They aren\'t always equal.'

代码实现结果如下：修改后元素不相同了