contrib.framework.remove_squeezable_dimensions
tf.contrib.framework.remove_squeezable_dimensions
tf.contrib.framework.remove_squeezable_dimensions
remove_squeezable_dimensions( predictions, labels, name=None )
Defined in tensorflow/contrib/framework/python/framework/tensor_util.py
.
See the guide: Framework (contrib)
Squeeze last dim if ranks of predictions
and labels
differ by 1.
This will use static shape if available. Otherwise, it will add graph operations, which could result in a performance hit.
Args:
-
predictions
: Predicted values, aTensor
of arbitrary dimensions. -
labels
: Label values, aTensor
whose dimensions matchpredictions
. -
name
: Name of the op.
Returns:
Tuple of predictions
and labels
, possibly with last dim squeezed.
© 2017 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/framework/remove_squeezable_dimensions