代码阅读记录

cls_mask = image_pred_ * (image_pred_[:, -1] == cls).float().unsqueeze(1)

首先image_pred_是个二维张量（即矩阵），-1指是是该数字所在位置的维度的最后一项。因此image_pred_[:, -1]表示这个矩阵中的列向量中的最后一列。多维张量也是如此，这个-1处在哪个位置，就是取了这个维度的最后一项。

(image_pred_[:, -1] == cls）这是个判断语句，首先image_pred_是个矩阵，需要判断的最后一列的列向量中的每个值，如果和cls相等，则返回true，不相等返回false，此时得到是个一维张量。这里除了==，还可以是>、<。

.float()可以将上面得到的true和false转变为1和0，并且为float格式。

unsqueeze(1)增加了一个维度，转换为二维张量，形状为（n*1).

此时再与image_pred_相乘，相当于将不是cls部分整一行全部变为了0。