TensorFlow 是一个用于数据流编程的开源软件库,用于数值计算。以下是一些官方文档的概览:

快速开始

安装

要开始使用 TensorFlow,您需要先安装它。以下是一个简单的步骤:

  1. 安装 Python:确保您的系统上安装了 Python 3.6 或更高版本。
  2. 安装 TensorFlow:使用 pip 安装 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], [1, 2, 3, 4], epochs=100)

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

资源

TensorFlow Logo