代码阅读记录
cls_mask = image_pred_ * (image_pred_[:, -1] == cls).float().unsqueeze(1)
首先image_pred_是个二维张量(即矩阵),-1指是是该数字所在位置的维度的最后一项。因此image_pred_[:, -1]表示这个矩阵中的列向量中的最后一列。多维张量也是如此,这个-1处在哪个位置,就是取了这个维度的最后一项。
(image_pred_[:, -1] == cls)这是个判断语句,首先image_pred_是个矩阵,需要判断的最后一列
原创
2021-06-02 16:35:59 ·
133 阅读 ·
0 评论