def nopeak_mask(self, size):
    np_mask = np.triu(np.ones((1, size, size)), k=1).astype('uint8')
    np_mask = Variable((torch.from_numpy(np_mask) == 0)).cuda()
    return np_mask
def nopeak_mask(self, size):
    np_mask = np.triu(np.ones((1, size, size)), k=1).astype('uint8')
    np_mask = Variable((torch.from_numpy(np_mask) == 0)).cuda()
    return np_mask