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.