Change TimeDistributedDense layer to TimeDistributed(Dense(...))

[JavierRoman]

Hi! I'm using latest version of keras and as TimeDistributedDense was removed in keras 2.0.0 i'd like to replace it with TimeDistributed + Dense.

I tried to replace:

model.add(TimeDistributedDense(input_dim=num_frequency_dimensions, output_dim=num_hidden_dimensions))

shown in cell 25 of the provided ipynb with:

model.add(TimeDistributed(Dense(num_hidden_dimensions), input_shape=(num_frequency_dimensions,)))

but i'm getting an assertation error. Any idea?