torch.Tensor.unfold¶
- Tensor.unfold(dimension, size, step) Tensor ¶
返回原始张量的视图,其中包含来自
self
张量在维度dimension
中的所有大小为size
的切片。两个切片之间的步长由
step
给定。如果 sizedim 是
self
的维度dimension
的大小,则返回的张量中维度dimension
的大小将为 (sizedim - size) / step + 1。在返回的张量中附加一个大小为
size
的额外维度。示例
>>> 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.]])