TensorFlow 2.0 RC is available Learn more

tf.contrib.framework.remove_squeezable_dimensions

View source on GitHub

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

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

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.