Difference between numpy dot() and Python 3.5+ matrix multiplication @

by

The @ operator calls the array’s __matmul__ method, not dot. This method is also present in the API as the function np.matmul.

>>> a = np.random.rand(8,13,13)
>>> b = np.random.rand(8,13,13)
>>> np.matmul(a, b).shape
(8, 13, 13)

From the documentation:

matmul differs from dot in two important ways.

  • Multiplication by scalars is not allowed.
  • Stacks of matrices are broadcast together as if the matrices were elements.

The last point makes it clear that dot and matmul methods behave differently when passed 3D (or higher dimensional) arrays. Quoting from the documentation some more:

For matmul:

If either argument is N-D, N > 2, it is treated as a stack of matrices residing in the last two indexes and broadcast accordingly.

For np.dot:

For 2-D arrays it is equivalent to matrix multiplication, and for 1-D arrays to inner product of vectors (without complex conjugation). For N dimensions it is a sum product over the last axis of a and the second-to-last of b

More Related Contents:

  1. What is the ‘@=’ symbol for in Python?
  2. how does multiplication differ for NumPy Matrix vs Array classes?
  3. Why list comprehension is much faster than numpy for multiplying arrays?
  4. How to get faster code than numpy.dot for matrix multiplication?
  5. How to get element-wise matrix multiplication (Hadamard product) in numpy?
  6. Multiple matrix multiplication
  7. What are the advantages of NumPy over regular Python lists?
  8. How does numpy.newaxis work and when to use it?
  9. What are type hints in Python 3.5?
  10. What is the difference between contiguous and non-contiguous arrays?
  11. Translate every element in numpy array according to key
  12. Automatically import modules when entering the python or ipython interpreter
  13. Numpy random choice to produce a 2D-array with all unique values
  14. Most efficient property to hash for numpy array
  15. Counting Cars OpenCV + Python Issue
  16. Vectorizing Haversine distance calculation in Python
  17. Parallelizing a Numpy vector operation
  18. How to install numpy and scipy for Ironpython27?
  19. Preserve custom attributes when pickling subclass of numpy array
  20. Scikit Learn SVC decision_function and predict
  21. How to run multiple commands synchronously from one subprocess.Popen command?
  22. How can I take the square root of -1 using python?
  23. Filtering a list based on a list of booleans
  24. Better way to shuffle two numpy arrays in unison
  25. Input and output numpy arrays to h5py
  26. Access n-th dimension in python [duplicate]
  27. How to use sklearn fit_transform with pandas and return dataframe instead of numpy array?
  28. numpy array TypeError: only integer scalar arrays can be converted to a scalar index
  29. Efficient dot products of large memory-mapped arrays
  30. mean from pandas and numpy differ

Leave a Comment