Pytorch:torch.diag()创建对角线张量方式

更新时间：2024年06月27日 15:06:06 作者：湫兮之风

这篇文章主要介绍了Pytorch:torch.diag()创建对角线张量方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

Pytorch torch.diag()创建对角线张量

torch.diag()

torch.diag是PyTorch中的一个函数，用于从给定的矩阵中提取对角线元素，或者构造一个以给定对角线元素为值的对角矩阵。这个函数对于矩阵分解和转换等操作非常重要。

如果输入是一个向量（1D张量），torch.diag会返回一个以该向量为对角线元素的2D方阵。如果输入是一个矩阵（2D张量），则返回一个包含输入矩阵对角线元素的1D张量。

torch.diag还允许你指定对角线的位置，通过参数diagonal实现。如果diagonal=0，则为主对角线；如果diagonal>0，则为位于主对角线之上的对角线；如果diagonal<0，则为位于主对角线之下的对角线。

语法：

input (Tensor)：输入张量。
diagonal (int, optional)：指定的对角线。
out (Tensor, optional)：输出张量。

举例一：

import torch
 
data = torch.tensor([1,2,3,4])
data_two = torch.diag(data,0)
print(data_two)

结果：

举例二：

import torch
 
data = torch.tensor(float('inf')).cuda().repeat(3)
data_two = torch.diag(data,0)
print(data_two)

结果：

torch.diag()取矩阵对角线元素,torch.diag_embed()指定值变成对角矩阵

1、torch.diag()

import torch
a = torch.randn(3, 3)
print(a)
tensor([[ 0.7594,  0.8073, -0.1344],
        [-1.7335, -0.4356, -0.0055],
        [ 1.8326,  0.3900, -0.9933]])
diag = torch.diag(a)  # 取 a 对角线元素，输出为 1*3
print(diag)
tensor([ 0.7594, -0.4356, -0.9933])

2、torch.diag_embed()

import torch
 
tensor([ 0.7594, -0.4356, -0.9933])
a_diag = torch.diag_embed(diag)  # 由 diag 变为三维 3*3
tensor([[ 0.7594,  0.0000,  0.0000],
        [ 0.0000, -0.4356,  0.0000],
        [ 0.0000,  0.0000, -0.9933]])

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

pandas创建series的三种方法小结
这篇文章主要介绍了pandas创建series的三种方法小结，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2022-05-05
Python清空文件并替换内容的实例
今天小编就为大家分享一篇Python清空文件并替换内容的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-10-10
python实现五子棋算法
这篇文章主要为大家详细介绍了python实现五子棋算法，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2022-05-05
使用python搭建服务器并实现Android端与之通信的方法
今天小编就为大家分享一篇使用python搭建服务器并实现Android端与之通信的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-06-06
Python 硬币兑换问题
这篇文章主要介绍了Python 硬币兑换问题，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-07-07
python xlwt如何设置单元格的自定义背景颜色
这篇文章主要介绍了python xlwt如何设置单元格的自定义背景颜色,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-09-09
python自定义线程池控制线程数量的示例
今天小编就为大家分享一篇python自定义线程池控制线程数量的示例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-02-02
Python模块_PyLibTiff读取tif文件的实例
今天小编就为大家分享一篇Python模块_PyLibTiff读取tif文件的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-01-01
在Python中实现字典反转案例
这篇文章主要介绍了在Python中实现字典反转案例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-12-12
Python使用enum模块获取枚举成员索引号的四种方法详解
在 Python 中,可以使用 enum 模块创建枚举类型,并通过遍历枚举成员来获取其索引号,本文介绍了常用的四种方法,大家可以根据自己的需要进行选择
2026-04-04