Question

我有以下功能：

def transpose_dot(vects):
    x, y = vects
    # <x,x> + <y,y> - 2<x,y>

    return K.dot(x, K.transpose(y))

当尝试使用keras进行评估时，它可以正常工作

x = K.variable(np.array(np_x))
y = K.variable(np.array(np_x))
obj = transpose_dot
objective_output = obj((x, y))
print('-----------------')
print (K.eval(objective_output))

结果：

[[ 1.  1.  1.  2.]
 [ 1.  2.  2.  4.]
 [ 1.  2.  2.  4.]
 [ 2.  4.  4.  8.]

但是，当尝试将其用作Lambda图层的功能时，它不起作用。

np_x = [[1, 0], [1, 1], [1, 1], [2, 2]]
features = np.array([np_x])
test_input = Input(shape=np.array(np_x).shape)
dot_layer= Lambda(transpose_dot, output_shape=(4,4))([test_input, test_input])
x = Model(inputs=test_input, outputs=dot_layer)
x.predict(features, batch_size=1)

结果

self.fn() if output_subset is None else\
ValueError: Shape mismatch: x has 2 cols (and 4 rows) but y has 4 rows (and 2 cols)
Apply node that caused the error: Dot22(Reshape{2}.0, Reshape{2}.0)
Toposort index: 11
Inputs types: [TensorType(float32, matrix), TensorType(float32, matrix)]
Inputs shapes: [(4, 2), (4, 2)]
Inputs strides: [(8, 4), (8, 4)]
Inputs values: ['not shown', 'not shown']
Outputs clients: [[Reshape{4}(Dot22.0, MakeVector{dtype='int64'}.0)]]

知道我在这里失踪的是什么吗？

编辑：添加了错误消息的输出

Answer 1

在https://github.com/fchollet/keras/的帮助下，我发现了自己的错误。函数期望得到（n，m）。但是当使用Lambda函数时，它期望获得（samples，n，m）。

Keras lambda函数点积错匹配

1 个答案: