jefflai108/Contrastive-Predictive-Coding-PyTorch

why torch.diag in nce loss?

artificertxj1 opened this issue · 0 comments

NVM, I mis-read the equation. You are right.