torch.vsplit

torch.vsplit(input, indices_or_sections) → List of Tensors

Разделяет input, тензор с двумя или более измерениями, на несколько тензоров вертикально в соответствии с indices_or_sections. Каждый сплит — это представление input.

Это эквивалентно вызову torch.tensor_split(input, indices_or_sections, dim=0) (размерность сплита — 0), за исключением того, что если indices_or_sections является целым числом, оно должно равномерно делиться на размерность сплита, иначе будет выброшено исключение во время выполнения.

Эта функция основана на функции NumPy numpy.vsplit().

Параметры:

• input (Тензор) – тензор для разделения.
• indices_or_sections (целое число или список или кортеж целых чисел) – См. аргумент в torch.tensor_split().

Пример::

>>> t = torch.arange(16.0).reshape(4,4)
>>> t
tensor([[ 0.,  1.,  2.,  3.],
        [ 4.,  5.,  6.,  7.],
        [ 8.,  9., 10., 11.],
        [12., 13., 14., 15.]])
>>> torch.vsplit(t, 2)
(tensor([[0., 1., 2., 3.],
         [4., 5., 6., 7.]]),
 tensor([[ 8.,  9., 10., 11.],
         [12., 13., 14., 15.]]))
>>> torch.vsplit(t, [3, 6])
(tensor([[ 0.,  1.,  2.,  3.],
         [ 4.,  5.,  6.,  7.],
         [ 8.,  9., 10., 11.]]),
 tensor([[12., 13., 14., 15.]]),
 tensor([], size=(0, 4)))