TensorFlow基本计算单元：代码示例

最新推荐文章于 2024-09-20 21:48:23 发布

@迷途小书童

最新推荐文章于 2024-09-20 21:48:23 发布

阅读量379

点赞数

分类专栏： TensorFlow框架学习

本文链接：https://blog.youkuaiyun.com/qq_43660987/article/details/92062065

版权

TensorFlow框架学习专栏收录该内容

8 篇文章

订阅专栏

本文通过一个具体的代码示例，展示了如何使用TensorFlow进行矩阵运算，包括变量的创建、初始化以及矩阵乘法的实现过程。

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

1，代码示例：

import tensorflow as tf
a = 3
#创建变量
w = tf.Variable([[0.6,1.2]])#创建行向量
x = tf.Variable([[2.1],[4.2]]) #创建列向量
y = tf.matmul(w, x) #矩阵相乘

init_op = tf.global_variables_initializer()# 初始化
with tf.Session() as sess:
   sess.run(init_op)
   print(y.eval()) #输出y的值

2，代码解析：

调用TensorFlow中相应的函数模块（API），需要将变量转换成TensorFlow支持的格式（tf.Variable），使用tf.Variable时，创建的是相应变量的内存区域（没有具体的值），在TensorFlow中需要进一步调用tf.global_variables_initializer，对全局的变量进行初始化操作（完成赋值操作），所有的上述操作需要在tf.Session定义的计算图中完成。