更新时间:2023-12-02 19:07:16
您可以组合Reshape和RepeatVector
Reshape
RepeatVector
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])
Keras-复制1D张量以创建3D张量
如何在Keras中使用“有状态"变量/张量创建自定义图层?