tf.compat.dimension_value
Stay organized with collections Save and categorize content based on your preferences.
Compatibility utility required to allow for both V1 and V2 behavior in TF.
tf.compat.dimension_value( dimension: Union['Dimension', int, None] ) -> Union[int, None]
Used in the notebooks
Until the release of TF 2.0, we need the legacy behavior of TensorShape
to coexist with the new behavior. This utility is a bridge between the two.
When accessing the value of a TensorShape dimension, use this utility, like this:
# If you had this in your V1 code: value = tensor_shape[i].value # Use `dimension_value` as direct replacement compatible with both V1 & V2: value = dimension_value(tensor_shape[i]) # This would be the V2 equivalent: value = tensor_shape[i] # Warning: this will return the dim value in V2!
Args |
dimension | Either a Dimension instance, an integer, or None. |
Returns |
A plain value, i.e. an integer or None. |