torch.Tensor.unfold

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

返回原始張量的檢視，該檢視包含在 dimension 維度上從 self 張量中提取的所有大小為 size 的切片。

兩個切片之間的步長由 step 指定。

如果 sizedim 是 self 張量在 dimension 維度的尺寸，則返回張量在 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.]])