Question

对于我的研究，我需要在2个密集层之间执行数学运算。也就是说，例如，取输出向量，将其加电2，然后将其传递给下一层。我尝试使用keras API，但这不起作用。例如：

inputOredData = Input(shape=(100,),name='pred')
powerOut = K.square (inputOredData )
prediction = Dense(2, activation='softmax')(powerOut )

我明白了：

You tried to call layer "dense_453". This layer has no information about its expected input shape, and thus cannot be built. You can build it manually via: `layer.build(batch_input_shape)`

任何帮助都将不胜感激。

Answer 1

您可以定义自己的图层，例如https://keras.io/layers/writing-your-own-keras-layers/

然后，为了您的目的，您可以按如下方式定义call（）方法：

def call(self, x, mask=None):
    return K.square(x)

Keras - 在图层之间添加数学运算

1 个答案: