在使用pytorch做矩陣的變換的時(shí)候可以使用這個(gè)函數(shù)，這個(gè)函數(shù)的目的在于返回矩陣上三角部分关霸，其余部分定義為0传黄。
方法的輸入有3個(gè)參數(shù)
input（輸入要進(jìn)行操作的tensor）
diagonal(int,optional)-表明要考慮哪個(gè)對(duì)角線。
| diagonal的取值 | 結(jié)果 |
|---|---|
| 空 | 輸入矩陣保留主對(duì)角線與主對(duì)角線以上的元素 |
| 正數(shù) n | 輸入矩陣保留主對(duì)角線與主對(duì)角線以上除去n行的元素（含對(duì)角線開始算往上） |
| 負(fù)數(shù) -n | 輸入矩陣保留主對(duì)角線與主對(duì)角線以上與主對(duì)角線下方h行對(duì)角線的元素 |
演示圖.png
