二值化生成mask: #通过阈值来二值化数组 w_mask = (self.word_mask > 0).int() 这是对于tensor的 或者 w_mask = np.int64(self.word_mask > 0)