as_strided

paddle. as_strided ( x, shape, stride, offset=0, name=None ) [源代码]

使用特定的 shape、stride、offset，返回 x 的一个 view Tensor。

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

参数

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

  • shape (list|tuple) - 指定的新的 shape。

  • stride (list|tuple) - 指定的新的 stride。

  • offset (int) - 指定的新的 offset。

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

返回

Tensor，x 的一个 view Tensor。

代码示例

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

>>> x = paddle.rand([2, 4, 6], dtype="float32")

>>> out = paddle.as_strided(x, [8, 6], [6, 1])
>>> print(out.shape)
[8, 6]
>>> # the stride is [6, 1].