strided_slice

paddle.strided_slice ( x, axes, starts, ends, strides, name ) [源代码]

strided_slice算子。

该OP沿多个轴生成 x 的切片，与numpy类似： https://docs.scipy.org/doc/numpy/reference/arrays.indexing.html。该OP使用 axes 、 starts 和 ends 属性来指定轴列表中每个轴的起点和终点位置，并使用此信息来对 x 切片。如果向 starts 或 ends 传递负值如

strided_slice - 图1

，则表示该轴的反向第 i−1i−1 个位置（这里以0为初始位置）， strides 表示切片的步长， strides 如果为负数，则按照反方向进行切片。如果传递给 starts 或 ends 的值大于n（维度中的元素数目），则表示n。当切片一个未知数量的维度时，建议传入 INT_MAX。 axes 、 starts 和 ends 以及 strides 四个参数的元素数目必须相等。以下示例将解释切片如何工作：

  1. 示例1：
  2.         给定：
  3.              data=[[1,2,3,4],[5,6,7,8],]
  4.              axes=[0,1]
  5.              starts=[1,0]
  6.              ends=[2,3]
  7.              strides=[1,1]
  9.         则：
  10.              result=[[5,6,7],]
  11. 示例2：
  12.         给定：
  13.              data=[[1,2,3,4],[5,6,7,8],]
  14.              axes=[0,1]
  15.              starts=[1,3]
  16.              ends=[2,0]
  17.              strides=[1,-1]
  19.         则：
  20.              result=[[8,7,6],]
  21. 示例3：
  22.         给定：
  23.              data=[[1,2,3,4],[5,6,7,8],]
  24.              axes=[0,1]
  25.              starts=[0,1]
  26.              ends=[-1,1000]    # 此处-1表示第0维的反向第0个位置，索引值是1。
  27.              strides =[1,3]
  28.         则：
  29.              result=[[2],]

参数：

  • x （Tensor）- 多维 Tensor，数据类型为 float32，float64，int32，或 int64。

  • axes （list|tuple）- 数据类型是 int32。表示进行切片的轴。

  • starts （list|tuple|Tensor）- 数据类型是 int32。如果 starts 的类型是 list 或 tuple，它的元素可以是整数或者形状为[1]的 Tensor。如果 starts 的类型是 Tensor，则是1-D Tensor。表示在各个轴上切片的起始索引值。

  • ends （list|tuple|Tensor）- 数据类型是 int32。如果 ends 的类型是 list 或 tuple，它的元素可以是整数或者形状为[1]的 Tensor。如果 ends 的类型是 Tensor，则是1-D Tensor。表示在各个轴上切片的结束索引值。

  • strides （list|tuple|Tensor）- 数据类型是 int32。如果 strides 的类型是 list 或 tuple，它的元素可以是整数或者形状为[1]的 Tensor。如果 strides 的类型是 Tensor，则是1-D Tensor。表示在各个轴上切片的步长。

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

返回：多维 Tensor，数据类型与 x 相同。

代码示例：

  1. import paddle
  3. x = paddle.zeros(shape=[3,4,5,6], dtype="float32")
  4. # example 1:
  5. # attr starts is a list which doesn't contain tensor Tensor.
  6. axes = [1, 2, 3]
  7. starts = [-3, 0, 2]
  8. ends = [3, 2, 4]
  9. strides_1 = [1, 1, 1]
  10. strides_2 = [1, 1, 2]
  11. sliced_1 = paddle.strided_slice(x, axes=axes, starts=starts, ends=ends, strides=strides_1)
  12. # sliced_1 is x[:, 1:3:1, 0:2:1, 2:4:1].
  13. # example 2:
  14. # attr starts is a list which contain tensor Tensor.
  15. minus_3 = paddle.full(shape=[1], fill_value=-3, dtype='int32')
  16. sliced_2 = paddle.strided_slice(x, axes=axes, starts=[minus_3, 0, 2], ends=ends, strides=strides_2)
  17. # sliced_2 is x[:, 1:3:1, 0:2:1, 2:4:2].