tf.keras.backend.squeeze
View source on GitHub |
Removes a 1-dimension from the tensor at index "axis".
tf.keras.backend.squeeze( x, axis )
Arguments | |
---|---|
x | A tensor or variable. |
axis | Axis to drop. |
Returns | |
---|---|
A tensor with the same data as x but reduced dimensions. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/keras/backend/squeeze