例如,来自
lens = [3, 5, 4]
我们想得到
mask = [[1, 1, 1, 0, 0], [1, 1, 1, 1, 1], [1, 1, 1, 1, 0]]
两者都是torch.LongTensors。
torch.LongTensor
nlp pytorch
nlp ×1
pytorch ×1