MoNCE
MoNCE copied to clipboard
Question about torch.einsum
Hello, authors!
I have diffculty understanding this line of code.
C = torch.einsum('bid,bod->bio', q, k)
What does torch.einsum do here? Could you provide a simple example? Thanks!