tensorflow.keras.layers.Add()的用法
时间: 2023-11-20 09:04:43 浏览: 414
tensorflow中keras支持的mnist数据集
在 TensorFlow 中,tf.keras.layers.Add() 是一个用于将多个张量相加的层。它的输入应该是一个张量列表,它们的形状应该相同,输出也是一个张量,形状与输入相同。
以下是 tf.keras.layers.Add() 的使用示例:
```python
import tensorflow as tf
# 定义两个 2x2 的张量
a = tf.constant([[1, 2], [3, 4]])
b = tf.constant([[5, 6], [7, 8]])
# 创建添加层
add_layer = tf.keras.layers.Add()
# 将两个张量相加
c = add_layer([a, b])
print(c)
```
输出结果为:
```
tf.Tensor(
[[ 6 8]
[10 12]], shape=(2, 2), dtype=int32)
```
在这个例子中,我们定义了两个 2x2 的张量 a 和 b,并创建了一个添加层 add_layer。然后,我们将张量 a 和 b 作为输入传递给 add_layer,并将其输出保存在 c 中。由于 a 和 b 的形状相同,因此它们可以相加,输出的形状也与输入相同。
阅读全文