Jump to content

NumPy: Difference between revisions

25 bytes added ,  6 February 2021
No edit summary
 
Line 16: Line 16:


===Batched Multiplication===
===Batched Multiplication===
If you need batched matrix multiplication, use [https://numpy.org/doc/stable/reference/generated/numpy.tensordot.html <code>np.tensordot</code>].
If you need batched matrix multiplication, use [https://numpy.org/doc/stable/reference/generated/numpy.tensordot.html <code>np.tensordot</code>] and transpose the result.