message: Combine into one block in example of docstring so that it could be rendered correctly
diff: @@ -1931,13 +1931,15 @@ def split(value, num_or_size_splits, axis=0, num=None, name="split"):
  For example:
  >>> x = tf.Variable(tf.random.uniform([5, 30], -1, 1))
  -
  - Split `x` into 3 tensors along dimension 1
  + >>>
  + >>> # Split `x` into 3 tensors along dimension 1:
  + ...
  >>> s0, s1, s2 = tf.split(x, num_or_size_splits=3, axis=1)
  >>> tf.shape(s0).numpy()
  array([ 5, 10], dtype=int32)
  -
  - Split `x` into 3 tensors with sizes [4, 15, 11] along dimension 1
  + >>>
  + >>> # Split `x` into 3 tensors with sizes [4, 15, 11] along dimension 1
  + ...
  >>> split0, split1, split2 = tf.split(x, [4, 15, 11], 1)
  >>> tf.shape(split0).numpy()
  array([5, 4], dtype=int32)
  