Skip to content

matmul2d

Multiplication for 2D matrices.

Parameters

  • A

  • B

Examples

>>> import pprint
>>> from river import utils

>>> A = {
...     (0, 0): 2, (0, 1): 0, (0, 2): 4,
...     (1, 0): 5, (1, 1): 6, (1, 2): 0
... }

>>> B = {
...     (0, 0): 1, (0, 1): 1, (0, 2): 0, (0, 3): 0,
...     (1, 0): 2, (1, 1): 0, (1, 2): 1, (1, 3): 3,
...     (2, 0): 4, (2, 1): 0, (2, 2): 0, (2, 3): 0
... }

>>> C = utils.math.matmul2d(A, B)
>>> pprint.pprint(C)
{(0, 0): 18.0,
    (0, 1): 2.0,
    (0, 2): 0.0,
    (0, 3): 0.0,
    (1, 0): 17.0,
    (1, 1): 5.0,
    (1, 2): 6.0,
    (1, 3): 18.0}