行列積 - kyopro-notes

行列積

code:py

def matmul(l, r, mod=None):

assert len(l0) == len(r), 'matrix dimension mismatch'

dim_row = len(l)

dim_col = len(r0)

ret = [0 * dim_col for _ in range(dim_row)]

if mod is not None:

inner_prod = lambda row, col: sum(a * b for a, b in zip(row, col)) % mod

else:

inner_prod = lambda row, col: sum(a * b for a, b in zip(row, col))

for i, row in enumerate(l):

for j, col in enumerate(zip(*r)):

retij = inner_prod(row, col)

return ret