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])
我有一个张量,我尝试将灰度图像堆叠 3 次到深度。
tf.stack([mask_tensor, mask_tensor, mask_tensor], 2)
如果我尝试这样做,总会出现另一个维度...
我已经在这个 post:
但他们不复制值,只是用零填充。
您可以使用tf.tile
,将输入张量沿第 3 维平铺 3 次:
tf.tile(mask_tensor, [1, 1, 3])