__init__.py « LSTM_unroll « examples « TensorFlowPythonExamples « res - platform/core/ml/nnfw - Domain: Machine Learning / ML Framework; Licenses: Apache-2.0;

index : platform/core/ml/nnfw

Domain: Machine Learning / ML Framework; Licenses: Apache-2.0;

Sangmin Seo <sangmin7.seo@samsung.com>, Sung-Jae Lee <sjlee@mail.com>

summary refs log tree commit diff

path: root/res/TensorFlowPythonExamples/examples/LSTM_unroll/__init__.py

blob: d21c2b54cc9b5838f6060dce06c5cee61e32645d (plain)

# 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, unroll=True))

# NOTE refer https://github.com/Samsung/ONE/issues/9895#issuecomment-1288436802