Tensorflow中矩阵运算函数

最新推荐文章于 2022-03-14 17:57:56 发布

昆仑-郑教主

最新推荐文章于 2022-03-14 17:57:56 发布

阅读量8.8k

点赞数 1

分类专栏： tensorflow 应用

本文链接：https://blog.csdn.net/zSean/article/details/75154118

版权

本文介绍了TensorFlow中几个关键的矩阵运算函数，包括tf.diag用于生成对角矩阵，tf.trace用于计算对角元素之和，tf.transpose实现维度调换，以及tf.cholesky进行 cholesky 分解。这些函数在处理矩阵和张量时非常实用。

摘要由CSDN通过智能技术生成

tf.diag(diagonal,name=None) #生成对角矩阵

import tensorflowas tf;
diagonal=[1,1,1,1]
with tf.Session() as sess:
     print(sess.run(tf.diag(diagonal)))

 #输出的结果为[[1 0 0 0]
               [0 1 0 0]
               [0 0 1 0]
               [0 0 0 1]]

tf.diag_part(input,name=None) #功能与tf.diag函数相反,返回对角阵的对角元素

import tensorflow as tf;
diagonal =tf.constant([[1,0,0,0],[0,1,0,0],[0,0,1,0],[0,0,0,1]])
with tf.Session() as sess:
    print(sess.run(tf.diag_part(diagonal)))

#输出结果为[1,1,1,1]

tf.trace(x,name=None) #求一个2维Tensor足迹，即为对角值diagonal之和

import tensorflow as tf;
diagonal =tf.constant([[1,0,0,3],[0,1,2,0],[0,1,1,0],[1,0,0,1]])
with tf.Session() as sess:
    print(sess.run(tf.trace(diagonal))