torch.triu
torch.triu(input, diagonal=0, out=None) → Tensor 返回矩阵上三角部分参数:input (Tensor) – the input tensordiagonal (int, optional) – the diagonal to considerout (Tensor, optional) – the output tensor如果diagonal为空,输出为原矩阵保留主对角线与主对角线以上的元素;如果diagonal为正数n,,输出为原矩阵保..