TensorFlow 2语法
时间: 2023-05-27 09:01:09 浏览: 48
TensorFlow 2是一个非常强大的深度学习框架,它提供了许多工具和语法来构建和训练人工神经网络。下面是一些常用的TensorFlow 2语法:
1. 导入TensorFlow 2库
```python
import tensorflow as tf
```
2. 定义一个变量
```python
x = tf.Variable(5.0)
```
3. 创建一个常量张量
```python
y = tf.constant(10.0)
```
4. 创建一个计算图
```python
@tf.function
def my_function(x):
z = x * y
return z
```
5. 执行计算图
```python
result = my_function(x)
```
6. 定义一个模型
```python
model = tf.keras.Sequential([tf.keras.layers.Dense(10, input_shape=(784,)),
tf.keras.layers.Dense(10)])
```
7. 编译模型
```python
model.compile(optimizer='adam',
loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),
metrics=['accuracy'])
```
8. 训练模型
```python
model.fit(x_train, y_train, epochs=10)
```
9. 评估模型
```python
model.evaluate(x_test, y_test, verbose=2)
```
10. 保存和加载模型
```python
# 保存模型
model.save('my_model.h5')
# 加载模型
loaded_model = tf.keras.models.load_model('my_model.h5')
```