tf.layers.dense tf2
时间: 2023-10-15 16:31:19 浏览: 153
tf2
在 TensorFlow 2 中,`tf.layers.dense` 被替代为 `tf.keras.layers.Dense`。`tf.keras.layers.Dense` 是 TensorFlow 中的全连接层,它将输入数据与权重矩阵相乘并添加偏置来生成输出。
下面是一个使用 `tf.keras.layers.Dense` 的示例:
```python
import tensorflow as tf
# 定义一个全连接层
dense_layer = tf.keras.layers.Dense(units=64, activation='relu')
# 定义输入数据
inputs = tf.random.normal(shape=(10, 32))
# 通过全连接层进行前向传播
outputs = dense_layer(inputs)
print(outputs.shape) # 输出为 (10, 64)
```
在上面的示例中,我们首先创建了一个 `Dense` 层,指定了输出单元的数量为 64,并使用 ReLU 激活函数。然后,我们创建了一个随机输入数据 `inputs`,其形状为 (10, 32)。最后,我们通过全连接层将输入数据传递进去,得到了输出 `outputs`。输出的形状为 (10, 64)。
阅读全文