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