diff options
Diffstat (limited to 'res/TensorFlowPythonExamples/examples/LSTM_retseq/__init__.py')
-rw-r--r-- | res/TensorFlowPythonExamples/examples/LSTM_retseq/__init__.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/res/TensorFlowPythonExamples/examples/LSTM_retseq/__init__.py b/res/TensorFlowPythonExamples/examples/LSTM_retseq/__init__.py new file mode 100644 index 000000000..2748c8096 --- /dev/null +++ b/res/TensorFlowPythonExamples/examples/LSTM_retseq/__init__.py @@ -0,0 +1,10 @@ +# NOTE tested with TF 2.8.0 +from tensorflow import keras + +model = keras.Sequential() +shape = (4, 4) + +model.add(keras.layers.InputLayer(input_shape=shape, batch_size=1)) +model.add(keras.layers.LSTM(2, input_shape=shape, return_sequences=True)) + +# NOTE refer https://github.com/Samsung/ONE/issues/9895#issuecomment-1289768739 |