快捷方式

torch.Tensor.unfold

Tensor.unfold(dimension, size, step) Tensor

返回原始张量的视图,其中包含来自 self 张量在维度 dimension 中的所有大小为 size 的切片。

两个切片之间的步长由 step 给定。

如果 sizedimself 的维度 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.]])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源