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 torchtensor([ 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]])

torch.diag() 取矩阵对角线元素，torch.diag_embed() 指定值变成对角矩阵相关推荐

LeetCode简单题之矩阵对角线元素的和
题目给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和. 请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和. 示例 1: 输入:mat = [[1,2,3], [4,5,6], ...
ACMNO.19 C语言-对角求和求一个3×3矩阵对角线元素之和。输入矩阵输出主对角线副对角线元素和样例输入 1 2 3 1 1 1 3 2 1 样例输出 3 7
这个我做了改进,可以实现NXN的矩阵.求出对角线之和! 只需要修改定义的define z的值就好! 接下来,进入正题! 题目描述: 求一个3×3矩阵对角线元素之和. 输入矩阵输出主对角线副对角 ...
C语言实验——求一个3*3矩阵对角线元素之和
C语言实验--求一个3*3矩阵对角线元素之和 Time Limit: 1000 ms Memory Limit: 65536 KiB Problem Description 给定一个3*3的矩阵,请你 ...
Java黑皮书课后题第8章：*8.2（求矩阵对角线元素的和）使用下面的方法头编写一个方法，求n*n的double类型矩阵中主对角线上所有数字的和。编写一个程序，读取一个4*4的矩阵，显示主对角线和
*8.2(求矩阵对角线元素的和)使用下面的方法头编写一个方法,求n*n的double类型矩阵中主对角线上所有数字的和.编写一个程序,读取一个4*4的矩阵,显示主对角线和题目题目描述与运行示例破题 ...
C语言学习之求一个3X3的整型矩阵对角线元素之和。
求一个3X3的整型矩阵对角线元素之和. int main(){int i,j,sum1=0,sum2=0; //定义 int a[3][3]; //定义数组 //给数组元素赋值 for(i=0;i&l ...
C语言学习之求一个3×3的整型矩阵对角线元素之和
求一个3×3的整型矩阵对角线元素之和一条对角线 #include <stdio.h> void main(){int a[3][3],i,j,s,s2;printf("请输入9 ...
C语言：求矩阵对角线元素的和
/*求矩阵对角线元素的和,以3阶矩阵为为例*/ #include <stdio.h> int main(){int a[3][3]={{1,2,3},{4,5,6},{7,8,9}};in ...
求一个3*3的整形矩阵对角线元素之和
/*求一个3*3的整形矩阵对角线元素之和*/#include <stdio.h>int main() {int a[3][3],i,j,sum = 0;printf("请输入整数 ...
python课程设计矩阵对角线之和_Python二维数组实现求出3*3矩阵对角线元素的和示例...
Python二维数组实现求出3*3矩阵对角线元素的和示例题目:求一个3*3矩阵对角线元素之和. 程序分析:利用双重for循环控制输入二维数组,再将a[i][i]累加后输出. def two_dime ...

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

1、torch.diag()

2、torch.diag_embed()

torch.diag() 取矩阵对角线元素，torch.diag_embed() 指定值变成对角矩阵相关推荐

最新文章

热门文章