MoNCE icon indicating copy to clipboard operation
MoNCE copied to clipboard

Question about torch.einsum

Open ShenZheng2000 opened this issue 2 years ago • 0 comments

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!

ShenZheng2000 avatar Feb 16 '23 19:02 ShenZheng2000