在使用pytorch做矩阵的变换的时候可以使用这个函数,这个函数的目的在于返回矩阵上三角部分,其余部分定义为0。
方法的输入有3个参数
input(输入要进行操作的tensor)
diagonal(int,optional)-表明要考虑哪个对角线。
diagonal的取值 | 结果 |
---|---|
空 | 输入矩阵保留主对角线与主对角线以上的元素 |
正数 n | 输入矩阵保留主对角线与主对角线以上除去n行的元素(含对角线开始算往上) |
负数 -n | 输入矩阵保留主对角线与主对角线以上与主对角线下方h行对角线的元素 |
演示图.png