diff --git a/tensorflow/python/ops/math_ops.py b/tensorflow/python/ops/math_ops.py index d5dfc646cd8ac3..c68c235828b1cb 100644 --- a/tensorflow/python/ops/math_ops.py +++ b/tensorflow/python/ops/math_ops.py @@ -640,7 +640,7 @@ def cast(x, dtype, name=None): ```python x = tf.constant([1.8, 2.2], dtype=tf.float32) - tf.cast(x, tf.int32) # [1, 2], dtype=tf.int32 + tf.dtypes.cast(x, tf.int32) # [1, 2], dtype=tf.int32 ``` The operation supports data types (for `x` and `dtype`) of