Tensorflow:将灰度图像(mxnx1)广播到(mxnx3)

Tensorflow: broadcast gray image (mxnx1) to (mxnx3)

我有一个张量,我尝试将灰度图像堆叠 3 次到深度。

tf.stack([mask_tensor, mask_tensor, mask_tensor], 2)

如果我尝试这样做,总会出现另一个维度...

我已经在这个 post:

但他们不复制值,只是用零填充。

您可以使用tf.tile,将输入张量沿第 3 维平铺 3 次:

tf.tile(mask_tensor, [1, 1, 3])