This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import tensorflow as tf | |
import numpy as np | |
if __name__ == '__main__': | |
np.random.seed(1) | |
# the size of the hidden state for the lstm (notice the lstm uses 2x of this amount so actually lstm will have state of size 2) | |
size = 1 | |
# 2 different sequences total | |
batch_size= 2 | |
# the maximum steps for both sequences is 10 |