5,323
edits
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. |