Tensorflow：廣播灰度圖像（mxnx1）到（mxnx3）

Question

我有一個張量，我嘗試將灰度圖像疊加3次到深度。

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

如果我試試這個，總會出現另一個維度......

我已經對這個職位：How to explicitly broadcast a tensor to match another's shape in tensorflow?

但他們不復制值，只是填補了零。

Answer 1

您可以使用tf.tile，沿第三維平鋪輸入張量的3倍：

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