W3cubDocs

/TensorFlow Python

tf.contrib.framework.remove_squeezable_dimensions(predictions, labels, name=None)

tf.contrib.framework.remove_squeezable_dimensions(predictions, labels, name=None)

See the guide: Framework (contrib)

Squeeze last dim if ranks of predictions and labels differ by 1.

This will use static shape if available. Otherwise, it will add graph operations, which could result in a performance hit.

Args:

  • predictions: Predicted values, a Tensor of arbitrary dimensions.
  • labels: Label values, a Tensor whose dimensions match predictions.
  • name: Name of the op.

Returns:

Tuple of predictions and labels, possibly with last dim squeezed.

Defined in tensorflow/contrib/framework/python/framework/tensor_util.py.

© 2017 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/api_docs/python/tf/contrib/framework/remove_squeezable_dimensions