We use Python 3.6

go back Level: Advanced (img: AV / score: 4) level Bite 31. Matrix multiplication / @ operator

15 out of 16 users completed this Bite ...
Will you be Pythonista #16 to crack this Bite?
Average Bite difficulty 1-10 rating: 9.0
» Up for a challenge? 💪

Since 3.5 Python has a binary operator to be used for matrix multiplication: @, see PEP 465 -- A dedicated infix operator for matrix multiplication.

The @ sign can now be used on types implementing the __matmul__ special/magic/dunder method.

It is important to note that whilst this feature shipped in 3.5, none of the standard library builtin types have matrix multiplication implementations. So let's try to implement it on a custom type for this Bite.

Implement a simple class called Matrix that takes a list of lists in its constructor.

Implement the __matmul__, __rmatmul__ (reversed) and __imatmul__ (in place) dunder methods.

Yes, using numpy a np.dot(self, other) would suffice, but the point is to get you thinking about implementing @ yourself!

Here is a how matrix multiplication works:

A = [[1, 2],  [3, 4]]
B = [[11, 12], [13, 14]]

Doing A @ B would do the following multiplications:

[[1 * 11 + 2 * 13,   1 * 12 + 2 * 14],
[3 * 11 + 4 * 13,   3 * 12 + 4 * 14]]

See the tests for more info. Good luck!

Special thanks to Anthony Shaw for providing the idea and collaborating with us on this Bite!
Github login button