torch.triu 与 numpy.triu 函数

地球被支点撬走啦

已于 2022-03-31 17:11:35 修改

阅读量5.1k

点赞数 2

分类专栏： Pytorch Python 文章标签： python pytorch

于 2022-03-31 16:28:50 首次发布

本文链接：https://blog.csdn.net/Flag_ing/article/details/123876284

版权

Python 同时被 2 个专栏收录

26 篇文章

订阅专栏

Pytorch

19 篇文章

订阅专栏

本文详细介绍了numpy和PyTorch中的`triu`函数，用于创建上三角矩阵。通过示例展示了如何设置参数k来选择不同的对角线，从而得到不同的上三角部分，并提供了等效的Python和PyTorch代码示例进行对比。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

triu = triangle up （我猜的）

顾名思义，这个函数的作用相同，都是返回上三角矩阵，定义分别如下：

numpy.triu(m, k)
torch.triu(Tensor, diagonal)

这两个函数的对应参数都是相同的含义。m 代表要操作的矩阵，k 代表以哪条对角线作为上三角矩阵的对角线。k=0 代表主对角线，k 为正数则从主对角线开始向上数第 k 条，k 为负数则从主对角线开始向下数第 k 条。举例如下：

import torch
import numpy as np

# triangle up
matrix = np.ones((4, 4))
# 使用主对角线作为上三角矩阵的对角线
tri_matrix = np.triu(matrix, k=0)
print(tri_matrix)
# 输出：
[[1. 1. 1. 1.]
 [0. 1. 1. 1.]
 [0. 0. 1. 1.]
 [0. 0. 0. 1.]]

# 使用主对角线上面第二条对角线
tri_matrix = np.triu(matrix, k=2)
print(tri_matrix)
# 输出：
[[0. 0. 1. 1.]
 [0. 0. 0. 1.]
 [0. 0. 0. 0.]
 [0. 0. 0. 0.]]

# 使用主对角线下面第一条对角线
tri_matrix = np.triu(matrix, k=-1)
print(tri_matrix)
# 输出：
[[1. 1. 1. 1.]
 [1. 1. 1. 1.]
 [0. 1. 1. 1.]
 [0. 0. 1. 1.]]

# torch 版
T_matrix = torch.from_numpy(matrix)
T_matrix = torch.triu(T_matrix, 1)
print(T_matrix)
# 输出：
tensor([[0., 1., 1., 1.],
        [0., 0., 1., 1.],
        [0., 0., 0., 1.],
        [0., 0., 0., 0.]], dtype=torch.float64)