目录

函数定义

参数

实例说明1：

实例说明2：

函数定义

torch.triu(input, diagonal=0, *,out=None) -> Tensor

返回input 张量，对应对角线 (diagonal）取值的结果。 其余位置为0

参数

input (Tensor): 输入张量

diagonal (int,可选的) - 要考虑的对角线

参数 diagonal 控制要考虑的对角线。如果diagonal = 0，则保留主对角线之上和之上的所有元素。其余位置的元素为0。正值不包括主对角线上方的对角线，类似地，负值包括主对角线下方的对角线。主对角线是i∈[0,min{d1,d2}−1] 的索引集{(i,i)}，其中d1,d2 是矩阵的维度。

out (Tensor, 可选的）- 输出张量

实例说明1：

import torch
t = torch.randn(3,3) # t reprsents a 3*3 dim  tensor
t

# result
tensor([[ 0.2133, -1.5566,  0.1254],
        [ 0.3418,  0.0389, -0.9250],
        [ 0.7646, -0.2630,  0.1733]])

torch. triu() # diagnoal默认值是0，即返回主对角线及其上三角的元素,其余元素为0

 torch.triu(t, diagnoal=1) # 返回主对角线上三角的元素，不包含主对角线上的元素。其余元素为0

 torch.triu(t, diagnoal=-1) #返回主对角线下一级以上所有元素。即不仅返回上三角（包含主对角线）的所有元素，还包含返回主对角线下一级的元素。其余元素为0

实例说明2：

import torch
t = torch.randn(4,6) # t reprsents a 3*3 dim  tensor
print(t)
# type(t)
#主对角线是i 属于 [0, min(d1,d2)-1]的索引值{（i,i）}
print(torch.triu(t)) # return 主对角线及其上三角的元素
print(torch.triu(t, diagonal=0))# return 主对角线及其上三角的元素
print(torch.triu(t, diagonal=1)) # return 主对角线上三角的元素，不包含主对角线上的元素
print(torch.triu(t, diagonal=-1)) # return 返回主对角线及其下方对角线以上的元素

运行结果：

最后

以上就是烂漫小蚂蚁为你收集整理的python torch.triu()函数函数定义的全部内容，希望文章能够帮你解决python torch.triu()函数函数定义所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。