我在损失函数中调用keras.backend.conv2d,当我尝试编译模型时,出现以下错误:
MXNet后端:无法自动推断卷积运算符的形状。请提供输入形状。给定输入形状-(无,无,无,无)
我知道我的输入张量的形状,因此,如果有某种方法可以手动指定我可以做到,但是在keras api的任何地方都找不到这样的东西。
答案 0 :(得分:1)
通常,您只需要将input_shape
添加为Conv2D
的参数。例如:
model.add(Conv2D(32, (3, 3), padding='same', input_shape=(256,256,3)))
如果您需要调用后端,请在调用keras.backend.conv2d()之前设置x._keras_shape =(256,256,3)。