torch.Tensor.unfold

Tensor.unfold(dimension, size, step) → Tensor

Возвращает представление исходного тензора, содержащее все срезы размера size из тензора self по измерению dimension.

Шаг между двумя срезами задаётся значением step.

Если sizedim — размер измерения dimension для self, размер измерения dimension в возвращаемом тензоре будет равен (sizedim - size) / step + 1.

В возвращаемый тензор добавляется дополнительное измерение размера size.

Параметры:

• dimension (int) – измерение, по которому выполняется разворачивание
• size (int) – размер каждого разворачиваемого среза
• step (int) – шаг между срезами

Пример:

>>> x = torch.arange(1., 8)
>>> x
tensor([ 1.,  2.,  3.,  4.,  5.,  6.,  7.])
>>> x.unfold(0, 2, 1)
tensor([[ 1.,  2.],
        [ 2.,  3.],
        [ 3.,  4.],
        [ 4.,  5.],
        [ 5.,  6.],
        [ 6.,  7.]])
>>> x.unfold(0, 2, 2)
tensor([[ 1.,  2.],
        [ 3.,  4.],
        [ 5.,  6.]])