model.py 文件源码

def _cosine_similarity_mask(self, simi):
        # [bsz, a_len, q_len]
        simi = torch.mul(simi, self.q_mask.unsqueeze(1).float()).clamp(min=eps)
        simi = torch.mul(simi, self.a_mask.unsqueeze(2).float()).clamp(min=eps)
        return simi