torch.einsum()

讲解
对比学习论文中出现:

# compute logits
# Einstein sum is more intuitive
# positive logits: Nx1
l_pos = torch.einsum('nc,nc->n', [q, k]).unsqueeze(-1)
# negative logits: NxK
l_neg = torch.einsum('nc,ck->nk', [q, self.queue.clone().detach()])
上一篇:GAN动漫人物头像生成


下一篇:pytorch einsum 矩阵乘 浅显易懂解释