tf.keras.backend.reshape
Stay organized with collections
Save and categorize content based on your preferences.
Reshapes a tensor to the specified shape.
tf.keras.backend.reshape(
x, shape
)
Arguments |
x
|
Tensor or variable.
|
shape
|
Target shape tuple.
|
Example:
a = tf.constant([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]])
a
<tf.Tensor: id=32, shape=(4, 3), dtype=int32, numpy=
array([[ 1, 2, 3],
[ 4, 5, 6],
[ 7, 8, 9],
[10, 11, 12]], dtype=int32)>
tf.keras.backend.reshape(a, shape=(2, 6))
<tf.Tensor: id=35, shape=(2, 6), dtype=int32, numpy=
array([[ 1, 2, 3, 4, 5, 6],
[ 7, 8, 9, 10, 11, 12]], dtype=int32)>