tf.contrib.framework.remove_squeezable_dimensions

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.