Если целое число, N, тензор будет разделен на N равных частей вдоль axis. Если одномерный массив отсортированных целых чисел, записи указывают индексы, на которых тензор будет разделен вдоль axis.
axis
Ось, вдоль которой следует разделить. По умолчанию 0.
Примечание
При использовании бэкенда Torch разделение не обязательно должно приводить к равному делению.
