Watch talks from the 2019 TensorFlow Dev Summit Watch now

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.