Reshape¶
- class
dragon.vm.tensorflow.keras.layers.
Reshape
(
target_shape,
**kwargs
)[source]¶ Layer to change the dimensions of input.
Examples:
x = tf.random.uniform((2, 1, 3)) # (2, 1, 3) => (2, 3) # Note that the dimensions should start from axis 1 print(tf.keras.layers.Reshape([3])(x)) # (2, 1, 3) => (2, 3) # At most one dimension could be set to ``-1`` # to infer the remain elements print(tf.keras.layers.Reshape([-1])(x)) # (2, 1, 3) => (2, 1, 3) # Set dimension to ``0`` will keep it unchanged print(tf.keras.layers.Reshape([0, -1])(x))