torch.hsplit

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

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

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

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

Параметры:

• input (Tensor) – тензор для разделения.
• indices_or_sections (int or list or tuple of ints) – см. аргумент в 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.hsplit(t, 2)
(tensor([[ 0.,  1.],
         [ 4.,  5.],
         [ 8.,  9.],
         [12., 13.]]),
 tensor([[ 2.,  3.],
         [ 6.,  7.],
         [10., 11.],
         [14., 15.]]))
>>> torch.hsplit(t, [3, 6])
(tensor([[ 0.,  1.,  2.],
         [ 4.,  5.,  6.],
         [ 8.,  9., 10.],
         [12., 13., 14.]]),
 tensor([[ 3.],
         [ 7.],
         [11.],
         [15.]]),
 tensor([], size=(4, 0)))