tf.contrib.framework.remove_squeezable_dimensions( predictions, labels, name=None )
See the guide: Framework (contrib)
Squeeze last dim if ranks of
labels differ by 1. (deprecated)
THIS FUNCTION IS DEPRECATED. It will be removed in a future version. Instructions for updating: Please switch to remove_squeezable_dimensions from tf.confusion_matrix. Note that the order of the inputs and outputs of labels and predictions have also been switched.
This will use static shape if available. Otherwise, it will add graph operations, which could result in a performance hit.
predictions: Predicted values, a
Tensorof arbitrary dimensions.
labels: Label values, a
Tensorwhose dimensions match
name: Name of the op.
labels, possibly with last dim squeezed.