我只是使用TensorFlow来实现CNN模型。在这个模型中,我需要连接两个4-D张量:形状为[16,128,128,3]的张量A和形状为[16,128,128,3]的张量B(16是批处理大小,128是图像块大小,3是频道数)。串联结果应为张量为C的张量为[16,128,128,6]。
我知道我们可以使用'tf.concat'函数来实现这一点,但是,该函数确实复制张量A和张量B,并且使用了较大的GPU内存。
如何在不使用'tf.concat'的情况下在TensorFlow中实现串联?
谢谢!