numpy 的triu及trul 函数以及参数k的解释

import numpy as np  

np.triu np.tril

例如

q=np.triu(np.ones((5,5)), k=0)

上面两个函数分别是返回上三角和下三角矩阵的函数，值得说的是参数k。  k表示从第几条对角线起保留数据。 正值是主对角线往上数，负值 是往下数。k=0时表示从主对角线开始保留。下面以5*5矩阵及triu 函数为例，不同的k参数的结果：

k=0

[[1. 1. 1. 1. 1.]
 [0. 1. 1. 1. 1.]
 [0. 0. 1. 1. 1.]
 [0. 0. 0. 1. 1.]
 [0. 0. 0. 0. 1.]]

k=1
[[0. 1. 1. 1. 1.]
 [0. 0. 1. 1. 1.]
 [0. 0. 0. 1. 1.]
 [0. 0. 0. 0. 1.]
 [0. 0. 0. 0. 0.]]

k=2
[[0. 0. 1. 1. 1.]
 [0. 0. 0. 1. 1.]
 [0. 0. 0. 0. 1.]
 [0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0.]]

k=3
[[0. 0. 0. 1. 1.]
 [0. 0. 0. 0. 1.]
 [0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0.]]

k=-1
[[1. 1. 1. 1. 1.]
 [1. 1. 1. 1. 1.]
 [0. 1. 1. 1. 1.]
 [0. 0. 1. 1. 1.]
 [0. 0. 0. 1. 1.]]

k=-2
[[1. 1. 1. 1. 1.]
 [1. 1. 1. 1. 1.]
 [1. 1. 1. 1. 1.]
 [0. 1. 1. 1. 1.]
 [0. 0. 1. 1. 1.]]

 

 

torch中也有类似函数，函数 为

 

torch.triu(input, diagonal=0）

 

diagonal 的意义与numpy中对应函数的k参数的意义一致。