torch:pad-sequence

(torch:pad-sequence sequences &key padding-value)

可変長のシーケンス (リスト、インデックスベクトル、テンソル) のリストを、パディング済みのランク 2 テンソル 1 つ、すなわちバッチ先頭の (batch 最長) として返します。各行は最長のものに合わせて padding-value (既定は 0) で埋められます。batch_first=True を指定した torch.nn.utils.rnn.pad_sequence に相当します。

結果はトークンインデックスの定数テンソルで、torch:embedding や torch:padding-mask にそのまま渡せます。