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!