開始使用免費開始

為自注意力設計遮罩

為了讓解碼器能學會預測權杖,在對輸入序列建模時,必須對未來的權杖進行遮罩。你將建立一個由 TrueFalse 組成的三角矩陣作為遮罩,並在右上三角(上對角)位置放入 False,以排除未來的權杖。

本練習屬於課程

Transformer Models with PyTorch

檢視課程

練習說明

  • 建立一個布林矩陣 tgt_mark,用來在解碼器主體的注意力機制中遮罩未來的權杖。

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

seq_length= 3

# Create a Boolean matrix to mask future tokens
tgt_mask = (1 - torch.____(
  torch.____(1, ____, ____), diagonal=____)
).____()

print(tgt_mask)
編輯並執行程式碼