Jacob B
Jacob B

Reputation: 724

Creating a Keras function

When creating a Keras function with the new tf2 I get a strange error that appears to be a bug, as I have used this same code before with a different model architecture, and that should not effect this.

Details on the model shapes are a (N,M,X) input and (N,M) output.

enter image description here

Upvotes: 0

Views: 38

Answers (1)

Jacob B
Jacob B

Reputation: 724

I solved this question by further specifying the input and output positions.

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

Upvotes: 1

Related Questions