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

9月前作者：清纯世纪分类：Toy博客阅读(38) 违法举报

这篇具有很好参考价值的文章主要介绍了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]])

文章来源地址https://www.toymoban.com/news/detail-602350.html

到了这里，关于torch.diag() 取矩阵对角线元素，torch.diag_embed() 指定值变成对角矩阵的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：如若内容造成侵权/违法违规/事实不符，请点击违法举报进行投诉反馈，一经查实，立即删除！

分享到：

领支付宝红包赞助服务器费用

1572. 矩阵对角线元素的和

给你一个正方形矩阵 mat，请你返回矩阵对角线元素的和。请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。同时求对角线和副对角线上元素的和再减去重合的元素

2024年02月13日
浏览(40)
【1572. 矩阵对角线元素的和】

来源：力扣（LeetCode）描述：给你一个正方形矩阵 mat ，请你返回矩阵对角线元素的和。请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。示例 1：示例 2：示例 3：提示： n == mat.length == mat[i].length 1 = n = 100 1 = mat[i][j] = 100 方法一：遍历矩阵思路

2024年02月12日
浏览(39)
C语言程序设计：求矩阵主对角线和副对角线元素之和

题目内容：求5行5列矩阵的主对角线和副对角线元素之和。输入格式: \\\"%d\\\" 输出格式： \\\"sum=%d\\\" 输入样例： 1 2 3 4 3 2 3 4 1 6 3 4 5 6 7 4 2 6 7 8 1 6 7 8 9 输出样例： sum=37 时间限制：500ms内存限制：32000kb

2024年02月13日
浏览(62)
Leetcode 1572.矩阵对角线元素之和

给你一个正方形矩阵 mat ，请你返回矩阵对角线元素的和。请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。示例 1：示例 2：示例 3：提示： n == mat.length == mat[i].length 1 = n = 100 1 = mat[i][j] = 100 通过次数 63.3K 提交次数 75.9K 通过率 83.3% 1.给一个

2024年02月10日
浏览(46)
求一个3*3矩阵对角线元素之和

#includestdio.h #includestring.h int main() { int m, n, j, k,sum=0; int a[10][10]; for (m = 0; m 3; m++) { for (n = 0; n 3; n++) { scanf(\\\"%d\\\", a[m][n]); } } for (m = 0; m 3; m++) { for (n = 0; n 3; n++) {

2024年02月03日
浏览(46)
LeetCode 1572. 矩阵对角线元素的和

力扣题目链接：https://leetcode.cn/problems/matrix-diagonal-sum/ 给你一个正方形矩阵 mat ，请你返回矩阵对角线元素的和。请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。示例 1：示例 2：示例 3：提示： n == mat.length == mat[i].length 1 = n = 100 1 = mat

2024年02月13日
浏览(65)
【LeetCode】1572.矩阵对角线元素的和

给你一个正方形矩阵 mat ，请你返回矩阵对角线元素的和。请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。示例 1：示例 2：示例 3：提示： n == mat.length == mat[i].length 1 = n = 100 1 = mat[i][j] = 100 观察矩阵的对角线，发现通常情况每一行都有两个

2024年02月13日
浏览(40)
题目：求一个3*3矩阵对角线元素之和

题目：求一个3*3矩阵对角线元素之和 There is no nutrition in the blog content. After reading it, you will not only suffer from malnutrition, but also impotence. The blog content is all parallel goods. Those who are worried about being cheated should leave quickly. 1.程序分析：利用双重for循环控制输入二维数组，再将a[i][i]累加

2024年04月17日
浏览(49)
【每日一题】1572. 矩阵对角线元素的和

给你一个正方形矩阵 mat，请你返回矩阵对角线元素的和。请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。示例 1：示例 2：示例 3：提示： n == mat.length == mat[i].length 1 = n = 100 1 = mat[i][j] = 100 思路：主对角线i=j，副对角线i+j=n-1。

2024年02月13日
浏览(52)
LeetCode解法汇总1572. 矩阵对角线元素的和

https://github.com/September26/java-algorithms 给你一个正方形矩阵 mat ，请你返回矩阵对角线元素的和。请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。示例 1：示例 2：示例 3：提示： n == mat.length == mat[i].length 1 = n = 100 1 = mat[i][j] = 100

2024年02月13日
浏览(32)