simCLR中的loss求解

图片:
simCLR的求解loss的代码，不理解最后两行，结合自己的理解和https://blog.csdn.net/weixin_40839812/article/details/127512686
写出了自己的理解如下：
首先logits的结构是第一列为和正例关系，其余列是和负例的关系，画图示意如下：

结合https://blog.csdn.net/weixin_40839812/article/details/127512686中的公式：

则最后一步推导
就可以理解了