TensorFlow 是一个开源的机器学习框架,由 Google Brain 团队开发。它广泛应用于各种机器学习和深度学习任务。本教程将为您介绍 TensorFlow 的基本概念和用法。

快速开始

以下是一个简单的 TensorFlow 示例,用于实现一个线性回归模型。

import tensorflow as tf

# 创建一个线性回归模型
model = tf.keras.Sequential([
    tf.keras.layers.Dense(units=1, input_shape=[1])
])

# 编译模型
model.compile(optimizer='sgd', loss='mean_squared_error')

# 训练模型
model.fit([1, 2, 3, 4], [1, 2, 3, 4], epochs=100)

# 预测
print(model.predict([5]))

安装 TensorFlow

要开始使用 TensorFlow,您需要先安装它。您可以通过以下命令安装:

pip install tensorflow

扩展阅读

更多关于 TensorFlow 的信息和教程,请访问 TensorFlow 官方文档

TensorFlow Logo