TensorFlow 是由 Google 开发并开源的机器学习与深度学习框架。
它最初为大规模分布式训练和推理设计,如今已成为 AI 研究和生产部署的主流工具之一。
TensorFlow 提供了灵活的 API,既能支持底层的数值计算图(Graph Execution),也能支持高层次的模型构建(如 Keras API)。
安装 :CPU 版本
pip install tensorflowGPU 版本(需 NVIDIA CUDA/cuDNN 环境):
pip install tensorflow-gpu常见应用场景:
(1)深度学习模型训练:图像识别、目标检测、自然语言处理、语音识别。
(2)机器学习研究:快速搭建原型,验证新模型结构。
(3)分布式训练:支持多 GPU、多 TPU 集群的并行计算。
(4)推理与部署:通过 TensorFlow Lite、TensorFlow.js、TensorFlow Serving 在移动端、浏览器和服务器上部署。
(5)强化学习与生成模型:可扩展到 RL、GAN、Transformers 等高级研究方向。
◆ ◆ ◆
核心概念
1、张量(Tensor)
TensorFlow 的基本数据结构,相当于多维数组。
张量可表示标量、向量、矩阵或高维数据。
2、计算图(Computation Graph)
TensorFlow 将运算组织为图结构,每个节点是操作(op),边表示数据流。
早期版本(TF1.x)需要显式构建图并在会话(Session)中运行;TF2.x 默认使用即时执行(Eager Execution),更贴近 Python 语义。
3、Keras 高层 API
tf.keras 提供了快速构建和训练模型的接口,是 TF2.x 的推荐方式。
4、设备与分布式支持
自动调度计算到 CPU/GPU/TPU。
提供 tf.distribute 策略,支持多机多卡。
5、模型保存与部署
SavedModel 格式支持完整模型(结构 + 参数)的保存与加载。
可导出到移动端或 Web。
◆ ◆ ◆
应用举例
例 1: 创建张量
print(tf.matmul(a, b)) # 矩阵乘法例 2:构建神经网络(Keras API)
print(model.summary())例 3:训练模型(以 MNIST 为例)
model.evaluate(x_test, y_test)例 4:保存与加载模型
new_model = tf.keras.models.load_model("my_model") 例 5: 使用 GPU
print("GPU available:", tf.config.list_physical_devices("GPU"))◆ ◆ ◆
常用模块速览
tf.constant / tf.Variable
创建张量与可训练参数。
tf.data
高效数据管道。
tf.keras
高层神经网络 API。
tf.nn
底层神经网络算子。
tf.linalg
矩阵与线性代数运算。
tf.distribute
分布式训练。
tf.saved_model
模型保存与加载。
tf.lite
移动端推理。
tfjs
Web 推理。
小结
TensorFlow 是一个全面的深度学习框架,既能提供底层数值计算支持,也能通过 Keras 进行高层建模。
它支持跨平台、分布式、高性能 的训练与部署,是科研与工业界广泛使用的工具。
无论是入门学习神经网络,还是构建大规模生产级 AI 系统,TensorFlow 都能提供可靠的支持。
![]()
“点赞有美意,赞赏是鼓励”
特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。
Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.