TF pre-2.0: defaultly static graph
TF2.0 alpha: defaultly dynamic graph
augmentation_layer=Sequential([
RandomFlip(mode='horizontal',name='random_horizontal_flip'),
RandomRotation( factor=0.1*2*math.pi,name='random_rotation' )
])
with tf.device('/cpu:0'): augim = augmentation_layer(image, training=True)