两个tensor合成一个tensor
时间: 2023-08-04 19:06:19 浏览: 116
两个Tensor可以通过TensorFlow的`concat`函数进行合并。如果想要两个Tensor上下合并,可以把`axis`设置为`0`。示例如下:
```python
import tensorflow as tf
# 创建两个Tensor
tensor1 = tf.constant([[1, 2, 3], [4, 5, 6]])
tensor2 = tf.constant([[7, 8, 9], [10, 11, 12]])
# 将两个Tensor上下合并
merged_tensor = tf.concat([tensor1, tensor2], axis=0)
# 创建 TensorFlow 会话
sess = tf.Session()
# 运行合并操作
result = sess.run(merged_tensor)
print(result)
```
在这个示例中,我们首先创建了两个形状相同的Tensor`tensor1`和`tensor2`,然后使用`concat`函数将两个Tensor在`axis=0`的维度上合并,得到一个形状为`(4, 3)`的Tensor`merged_tensor`。最后,我们创建了一个TensorFlow会话,并通过`run`方法运行合并操作,得到合并后的结果`result`。
如果想要将两个Tensor左右合并,可以把`axis`设置为`1`。例如:
```python
import tensorflow as tf
# 创建两个Tensor
tensor1 = tf.constant([[1, 2], [3, 4]])
tensor2 = tf.constant([[5, 6], [7, 8]])
# 将两个Tensor左右合并
merged_tensor = tf.concat([tensor1, tensor2], axis=1)
# 创建 TensorFlow 会话
sess = tf.Session()
# 运行合并操作
result = sess.run(merged_tensor)
print(result)
```
在这个示例中,我们首先创建了两个形状相同的Tensor`tensor1`和`tensor2`,然后使用`concat`函数将两个Tensor在`axis=1`的维度上合并,得到一个形状为`(2, 4)`的Tensor`merged_tensor`。最后,我们创建了一个TensorFlow会话,并通过`run`方法运行合并操作,得到合并后的结果`result`。
阅读全文