python - 创建 Keras 函数

Question

使用新的 tf2 创建 Keras 函数时，我收到一个奇怪的错误，这似乎是一个错误，因为我之前使用过相同的代码和不同的模型架构，这不应该影响这一点。

模型形状的详细信息是 (N,M,X) 输入和 (N,M) 输出。

标签： pythontensorflowkeras

Answer 1

我通过进一步指定输入和输出位置解决了这个问题。

K.function(model.layers[0].get_input_at(0), model.layers[-1].get_output_at(0))