norm = tf.keras.layers.BatchNormalization()(conv1)改为tensorflow2.0使用
时间: 2024-05-05 10:19:43 浏览: 71
TensorFlow实现Batch Normalization
在 TensorFlow 2.0 中,您可以使用 tf.keras.layers.BatchNormalization() 直接创建批归一化层,示例如下:
```python
norm = tf.keras.layers.BatchNormalization()(conv1)
```
改为:
```python
norm = tf.keras.layers.BatchNormalization()(inputs)
```
其中 `inputs` 是输入层的张量,您需要将其传递给 `BatchNormalization()` 层。
阅读全文