TensorFlow 是一个由 Google 开发的开源机器学习框架,用于构建和训练各种机器学习模型。本教程将为您介绍 TensorFlow 的基础知识。

快速开始

安装 TensorFlow

在开始之前,您需要安装 TensorFlow。您可以通过以下命令进行安装:

pip install 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, 5], [1, 2, 3, 4, 5], epochs=100)

# 使用模型进行预测
print(model.predict([6]))

深入学习

要深入了解 TensorFlow,您可以访问我们的 TensorFlow 教程

TensorFlow Logo


如果您有任何关于 TensorFlow 的问题,欢迎在 社区论坛 中提问。