如何在不使用'tf.concat'的情况下在TensorFlow中实现串联?

时间:2019-05-24 09:40:58

标签: python tensorflow

我只是使用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中实现串联?

谢谢!

0 个答案:

没有答案