[tf.data] Fix bug in `tf.data.experimental.unbatch()`.
Previously, if the rank of the input to this transformation was statically unknown, we would erroneously report that the output is a scalar, and violate downstream shape integrity checks. Instead, in that case the output shape should be unknown. PiperOrigin-RevId: 215683027
Loading
Please sign in to comment