Update tutorial_learning2learn.py

PennyLaneAI · Sep 13, 2023 · 52de7e6 · 52de7e6
1 parent 8a6d7c7
commit 52de7e6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/demonstrations/tutorial_learning2learn.py b/demonstrations/tutorial_learning2learn.py
@@ -896,10 +896,10 @@ def call(self, inputs):
 rnn0 = QRNN(graph=_graph)
 
 # Create some input layers to feed the data
-inp_cost = tf.keras.layers.Input(shape=(1,), dtype=tf.float64)
-inp_params = tf.keras.layers.Input(shape=(2,), dtype=tf.float64)
-inp_h = tf.keras.layers.Input(shape=(2,), dtype=tf.float64)
-inp_c = tf.keras.layers.Input(shape=(2,), dtype=tf.float64)
+inp_cost = tf.keras.layers.Input(shape=(1,))
+inp_params = tf.keras.layers.Input(shape=(2,))
+inp_h = tf.keras.layers.Input(shape=(2,))
+inp_c = tf.keras.layers.Input(shape=(2,))
 
 # Manually creating the recurrent loops. In this case just three iterations are used.
 out0 = rnn0([inp_cost, inp_params, inp_h, inp_c])