def __init__(self):
matrix_pairs = {}
rng = np.random.RandomState(SEED)
for sz in SIZES:
array = nla.qr(rng.standard_normal((sz, sz)))[0]
matrix_pairs[sz] = (matrices.OrthogonalMatrix(array), array)
super().__init__(matrix_pairs, rng)
def __init__(self):
super().__init__()
for sz in SIZES:
array = nla.qr(self.rng.standard_normal((sz, sz)))[0]
self.matrices[sz] = matrices.OrthogonalMatrix(array)
self.np_matrices[sz] = array
def matrix_pair(self, rng, size):
array = nla.qr(rng.standard_normal((size, size)))[0]
return matrices.OrthogonalMatrix(array), array