View source on GitHub |
Compatibility utility required to allow for both V1 and V2 behavior in TF.
tf.compat.dimension_value( dimension )
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!
Arguments | |
---|---|
dimension | Either a Dimension instance, an integer, or None. |
Returns | |
---|---|
A plain value, i.e. an integer or None. |
© 2020 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/versions/r2.4/api_docs/python/tf/compat/dimension_value