Python tf.compat.dimension_value用法及代码示例

允许 TF 中的 V1 和 V2 行为所需的兼容性实用程序。

用法

tf.compat.dimension_value(
    dimension
)

在 TF 2.0 发布之前，我们需要 TensorShape 的旧行为与新行为共存。该实用程序是两者之间的桥梁。

访问 TensorShape 维度的值时，请使用此实用程序，如下所示：

# 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!