向3通道张量添加一个附加通道

Question

我想将形状张量[310241024]和形状张量[110241024]组合起来，以形成单个形状张量[410241024]

这是为了将RGB图像的通道组合为每个像素的[r，g，b，d]格式的深度图像

我目前正在尝试这样做：

tensor = tf.concat([imageTensor, depthTensor], axis=2)

但是我收到错误

InvalidArgumentError: ConcatOp : Dimensions of inputs should match: shape[0] = [3,1024,1024] vs. shape[1] = [1,1024,1024] [Op:ConcatV2]

我只是想知道该怎么做？

Answer 1

您想要在轴=0:上连接

import tensorflow as tf
t1 = tf.random.uniform((3, 1024, 1024))
t2 = tf.random.uniform((1, 1024, 1024))
final_tensor = tf.concat((t1, t2), axis=0)
print(final_tensor.shape)
(4, 1024, 1024)