tensorflow: tf.repeat fails

Please make sure that this is a bug. As per our GitHub Policy, we only address code/doc bugs, performance issues, feature requests and build/installation issues on GitHub. tag:bug_template

System information

• Have I written custom code (as opposed to using a stock example script provided in TensorFlow): Yes
• OS Platform and Distribution (e.g., Linux Ubuntu 16.04): MacOS Mojave 10.14.6
• Mobile device (e.g. iPhone 8, Pixel 2, Samsung Galaxy) if the issue happens on mobile device: No
• TensorFlow installed from (source or binary): pip
• TensorFlow version (use command below): v2.4.0-rc4-71-g582c8d236cb 2.4.0
• Python version: 3.*
• Bazel version (if compiling from source): n/a
• GCC/Compiler version (if compiling from source): n/a
• CUDA/cuDNN version: n/a
• GPU model and memory: no GPU

You can collect some of this information using our environment capture script You can also obtain the TensorFlow version with:

1. TF 1.0: python -c "import tensorflow as tf; print(tf.GIT_VERSION, tf.VERSION)"
2. TF 2.0: python -c "import tensorflow as tf; print(tf.version.GIT_VERSION, tf.version.VERSION)"

Describe the current behavior

n = 50000
many_ones = tf.cast(tf.concat([tf.ones(n), [n,]], axis = 0), 'int32')
array = tf.concat([tf.range(n), [n, ]], axis = 0)
tf.repeat(array, many_ones)

returns

InvalidArgumentError: Size 0 must be non-negative, not -1794917296 [Op:Reshape]

Describe the expected behavior np.repeat returns

array([ 0, 1, 2, …, 50000, 50000, 50000], dtype=int32)

Standalone code to reproduce the issue Provide a reproducible test case that is the bare minimum necessary to generate the problem. If possible, please share a link to Colab/Jupyter/any notebook.

the following code would work for small ns, but fail/be very slow for big ns

n = 50000
many_ones = tf.cast(tf.concat([tf.ones(n), [n,]], axis = 0), 'int32')
array = tf.concat([tf.range(n), [n, ]], axis = 0)
tf.repeat(array, many_ones)

Other info / logs Include any logs or source code that would be helpful to diagnose the problem. If including tracebacks, please include the full traceback. Large logs and files should be attached.