tf.keras.ops.squeeze
Remove axes of length one from x.
tf.keras.ops.squeeze(
x, axis=None
)
| Args | |
|---|---|
x | Input tensor. |
axis | Select a subset of the entries of length one in the shape. |
| Returns | |
|---|---|
| The input tensor with all or a subset of the dimensions of length 1 removed. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/ops/squeeze
