jung hyemin
jung hyemin

Reputation: 675

What is the meaning of the output shape in model.summary of KERAS?

Below is the RNNs model.

enter image description here

And In keras, the output shape of the below code is (N,10).

model=Sequential()
model.add(RNN(10, input_shape=(1, look_back)))

I know that N in the output shape (N,10) is the batch_size.

What I want to know is that '10' is the enter image description here?

Upvotes: 0

Views: 561

Answers (1)

scarecrow
scarecrow

Reputation: 6874

Yes 10 is D_h. Basically RNN(units) denotes that the layer returns the vector h_t from the final timestep of the RNN with the size being the number you specify for the units parameter, which in this case is 10. Or in other words, units parameter denotes the dimensionality of the output vector of that particular layer.

Upvotes: 1

Related Questions