tf.contrib.framework.remove_squeezable_dimensions

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

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

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.

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.