Python Keras-复制一维张量以创建三维张量

我目前正在尝试设计一个生成器模型，它将输入图像和连接在图像中的条件标签结合起来

输入图像为3D（包括通道），标签为1D（包含类的idx）

根据的信息，我需要将

cond

重塑为与

input\u image

相同的形状。但据我在Keras文档中发现的，只有二维张量重复到三维张量

---

如何将

cond

输入复制为相同的形状，以便将其连接起来？

您可以将

重塑

和

重复向量

input_image = Input(shape=(row, col, chann))
cond = Input(shape=(1,))
cond = Reshape((row,col,1))(RepeatVector(row*col)(cond))
concat = Concatenate()([input_image, cond])

---

那么如何将二维张量复制成三维张量呢？比如复制一个热向量，使其成为图像的形状（行、列），以便将其连接起来？

input_image = Input(shape=(row, col, chann))
cond = Input(shape=(1,))
cond = Reshape((row,col,1))(RepeatVector(row*col)(cond))
concat = Concatenate()([input_image, cond])