Skip to content

Instantly share code, notes, and snippets.

@alexbrillant
Created October 18, 2020 20:06
Show Gist options
  • Save alexbrillant/c4d4670d9a5ab0832bde518b69fc4ee2 to your computer and use it in GitHub Desktop.
Save alexbrillant/c4d4670d9a5ab0832bde518b69fc4ee2 to your computer and use it in GitHub Desktop.
time_steps = 50
input_dim = 1
output_dim = 1
dataset = tf.data.Dataset.from_generator(
generator=generator,
output_types=(tf.float32, tf.float32),
output_shapes=(tf.TensorShape([None, time_steps, input_dim]), tf.TensorShape([None, time_steps, output_dim]))
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment