tf.experimental.numpy.flatten

TensorFlow variant of NumPy's flatten.