【机器学习】TensorFlow编程基础

news2024/11/14 4:08:54

🌈个人主页: 鑫宝Code
🔥热门专栏: 闲话杂谈｜炫酷HTML | JavaScript基础
💫个人格言: "如无必要，勿增实体"

文章目录

TensorFlow编程基础
- 张量(Tensor)
- 计算图(Computational Graph)
- 会话(Session)
- 基本数学运算

TensorFlow编程基础

TensorFlow是一个由Google Brain团队开发的开源人工智能框架,用于构建和训练深度学习模型。它具有强大的数值计算能力,并支持在CPU和GPU上高效运行。本文将介绍TensorFlow编程的基础知识,包括张量(Tensor)的概念、计算图(Computational Graph)、会话(Session)等核心组件,以及一些基本的数学运算。
在这里插入图片描述

张量(Tensor)

在TensorFlow中,数据被表示为张量(Tensor)的形式。张量是一个多维数组或列表,其中的元素可以是不同的数据类型,如浮点数、整数或字符串等。张量的阶数(rank)表示其维度的数量,形状(shape)则描述了每个维度的大小。

例如,一个标量(scalar)是一个0阶张量,一个向量(vector)是一个1阶张量,一个矩阵(matrix)是一个2阶张量,以此类推。

我们可以使用tf.constant()函数创建一个常量张量:
在这里插入图片描述

scalar = tf.constant(3.14)
vector = tf.constant([1.0, 2.0, 3.0])
matrix = tf.constant([[1.0, 2.0], [3.0, 4.0]])

计算图(Computational Graph)

TensorFlow使用计算图来表示数学计算过程。计算图是由一系列节点(Node)和边(Edge)组成的有向图,其中节点代表操作(Operation),边则表示操作之间的依赖关系。

在TensorFlow中,我们首先构建计算图,然后在会话(Session)中执行计算图中的操作。这种延迟执行(Deferred Execution)的方式使得TensorFlow能够优化计算过程,并支持在多种设备(如CPU、GPU或TPU)上高效运行。

下面是一个简单的计算图示例,它计算两个常量张量的和:
在这里插入图片描述

import tensorflow as tf

a = tf.constant(3.0)
b = tf.constant(4.0)
c = a + b

with tf.Session() as sess:
    result = sess.run(c)
    print(result)  # Output: 7.0

在上面的代码中,我们首先创建了两个常量张量a和b,然后定义了一个操作c来计算它们的和。接着,我们在会话(Session)中执行操作c,并获取计算结果。

会话(Session)

会话(Session)是TensorFlow中用于执行计算图的主要机制。它负责分配资源(如内存和计算设备),并在计算图中执行操作。

我们可以使用tf.Session()创建一个会话对象,然后调用其run()方法来执行计算图中的操作。run()方法接受一个或多个操作作为输入,并返回相应的计算结果。

import tensorflow as tf

x = tf.constant(2.0)
y = tf.constant(3.0)
z = x * y

with tf.Session() as sess:
    result = sess.run(z)
    print(result)  # Output: 6.0

在上面的示例中,我们创建了一个会话,并在其中执行了乘法操作z。

基本数学运算

TensorFlow提供了广泛的数学运算,包括基本的算术运算、矩阵运算、统计函数等。这些运算都是通过计算图中的操作节点来实现的。

下面是一些常见的数学运算示例:

算术运算

import tensorflow as tf

a = tf.constant(3.0)
b = tf.constant(4.0)

# 加法
c = tf.add(a, b)

# 减法
d = tf.subtract(b, a)

# 乘法
e = tf.multiply(a, b)

# 除法
f = tf.divide(b, a)

with tf.Session() as sess:
    print(sess.run(c))  # Output: 7.0
    print(sess.run(d))  # Output: 1.0
    print(sess.run(e))  # Output: 12.0
    print(sess.run(f))  # Output: 1.3333333

矩阵运算

import tensorflow as tf

a = tf.constant([[1.0, 2.0], [3.0, 4.0]])
b = tf.constant([[2.0, 1.0], [3.0, 2.0]])

# 矩阵乘法
c = tf.matmul(a, b)

with tf.Session() as sess:
    print(sess.run(c))
    # Output:
    # [[8. 5.]
    #  [18. 13.]]

统计函数

import tensorflow as tf

a = tf.constant([1.0, 2.0, 3.0, 4.0])

# 求和
sum_a = tf.reduce_sum(a)

# 平均值
mean_a = tf.reduce_mean(a)

with tf.Session() as sess:
    print(sess.run(sum_a))  # Output: 10.0
    print(sess.run(mean_a))  # Output: 2.5