小编047*_*037的帖子

如何在PyTorch中将句子长度批量转换为掩码?

例如,来自

lens = [3, 5, 4]
Run Code Online (Sandbox Code Playgroud)

我们想得到

mask = [[1, 1, 1, 0, 0],
        [1, 1, 1, 1, 1],
        [1, 1, 1, 1, 0]]
Run Code Online (Sandbox Code Playgroud)

两者都是torch.LongTensors。

nlp pytorch

2
推荐指数
2
解决办法
822
查看次数

标签 统计

nlp ×1

pytorch ×1