numpy Matrix Multiplication Simplification - is it possible?

Question

Is there a way to simplify

 a=np.dot(a,b)

just like the way you write a=a+b as a+=b ? (a,b are both np.array)

Answer 1

In Python3.5+ you can use the @ operator for matrix multiplication, e.g.:

import numpy as np

a = np.random.randn(4, 10)
b = np.random.randn(10, 5)

c = a @ b

This is equivalent to calling c = np.matmul(a, b). Inplace matrix multiplication (@=) is not yet supported (and doesn't make sense in most cases anyway, since the output usually has different dimensions to the first input).

Also note that np.matmul (and @) will behave differently to np.dot when one or more of the input arrays has >2 dimensions (see here).