description: TensorFlow variant of NumPy's vsplit.

tf.experimental.numpy.vsplit

View source on GitHub

TensorFlow variant of NumPy's vsplit.

tf.experimental.numpy.vsplit(
    ary, indices_or_sections
)

See the NumPy documentation for numpy.vsplit.