unfold¶

paddle. unfold ( x, axis, size, step, name=None ) [源代码] ¶

返回 x 的一个 view Tensor。以滑动窗口式提取 x 的值。

仅在动态图下可用，返回的 Tensor 和 x 共享内存。

参数¶

x (Tensor) - 输入多维 Tensor，可选的数据类型为 'float16'、'float32'、'float64'、'int16'、'int32'、'int64'、'bool'、'uint16'。

axis (int) - 表示需要提取的维度。如果 axis < 0，则维度为 rank(x) + axis。

size (int) - 表示需要提取的窗口长度。

step (int) - 表示每次提取跳跃的步长。

name (str，可选) - 具体用法请参见 Name，一般无需设置，默认值为 None。

返回¶

Tensor，x 的一个 view Tensor。

代码示例¶

          >>> import paddle
>>> paddle.base.set_flags({"FLAGS_use_stride_kernel": True})

>>> x = paddle.arange(9, dtype="float64")

>>> out = paddle.unfold(x, 0, 2, 4)
>>> print(out)
Tensor(shape=[2, 2], dtype=float64, place=Place(cpu), stop_gradient=True,
[[0., 1.],
 [4., 5.]])

         