tf.compat.dimension_value

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

Used in the tutorials

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!

dimension Either a Dimension instance, an integer, or None.

A plain value, i.e. an integer or None.