损失函数中的“ MXNet后端:无法自动推断卷积算子的形状”

时间:2018-07-24 21:12:33

标签: keras mxnet

我在损失函数中调用keras.backend.conv2d,当我尝试编译模型时,出现以下错误:

  

MXNet后端:无法自动推断卷积运算符的形状。请提供输入形状。给定输入形状-(无,无,无,无)

我知道我的输入张量的形状,因此,如果有某种方法可以手动指定我可以做到,但是在keras api的任何地方都找不到这样的东西。

1 个答案:

答案 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)。