tf.contrib.framework.remove_squeezable_dimensions

View source on GitHub

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

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

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

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