07 TensorFlow 2.0：RNNCell實戰1

import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'

import tensorflow as tf
from tensorflow.keras import  layers, optimizers, datasets, Sequential 

from tensorflow.python.client import  device_lib
print(tf.__version__)
print(device_lib.list_local_devices())
gpu_ok = tf.test.is_gpu_available()
print("\nuse GPU",gpu_ok)

import os
cell = layers.SimpleRNNCell(units=3, 
                            activation='tanh', use_bias=True, 
                            kernel_initializer='glorot_uniform', 
                            recurrent_initializer='orthogonal', 
                            bias_initializer='zeros', 
                            kernel_regularizer=None, 
                            recurrent_regularizer=None, 
                            bias_regularizer=None, 
                            kernel_constraint=None, 
                            recurrent_constraint=None, 
                            bias_constraint=None, 
                            dropout=0.0, 
                            recurrent_dropout=0.0)

cell.build(input_shape=(None,4))
cell.trainable_variables