linalg:triu

(linalg:triu array &key k)

array の上三角を返します。k 番目の対角線より下をすべて 0 にしたコピーです (numpy の np.triu)。:k のデフォルトは 0 で主対角線を残します。正の :k は境界を右上へ、負の :k は左下へ動かします。rank は 2 以上である必要があり、行列のスタックの場合は最後の 2 軸が行列とみなされます。すべて 1 の行列に :k 1 で適用すると causal (subsequent) アテンションマスクになります。対になるのは linalg:tril です。