Back{: .button}
The Einstein notation is available in numpy as np.einsum
2D matrix multiplication can be represented as
np.einsum("ik,kj->ij", A,B)When using cupy.einsum the operation can utilise tensor cores in the GPU
Back{: .button}
The Einstein notation is available in numpy as np.einsum
2D matrix multiplication can be represented as
np.einsum("ik,kj->ij", A,B)When using cupy.einsum the operation can utilise tensor cores in the GPU