Pytorch-triu

在使用pytorch做矩阵的变换的时候可以使用这个函数,这个函数的目的在于返回矩阵上三角部分,其余部分定义为0。

方法的输入有3个参数

input(输入要进行操作的tensor)

diagonal(int,optional)-表明要考虑哪个对角线。

diagonal的取值 结果
输入矩阵保留主对角线与主对角线以上的元素
正数 n 输入矩阵保留主对角线与主对角线以上除去n行的元素(含对角线开始算往上)
负数 -n 输入矩阵保留主对角线与主对角线以上与主对角线下方h行对角线的元素
演示图.png

你可能感兴趣的:(Pytorch-triu)