slice
paddle.fluid.layers.
slice
(input, axes, starts, ends)[源代码]
该OP沿多个轴生成 input
的切片。与numpy类似: https://docs.scipy.org/doc/numpy/reference/arrays.indexing.html 该OP使用 axes
、 starts
和 ends
属性来指定轴列表中每个轴的起点和终点位置,并使用此信息来对 input
切片。如果向 starts
或 ends
传递负值如
,则表示该轴的反向第
个位置(这里以0为初始位置)。如果传递给 starts
或 end
的值大于n(维度中的元素数目),则表示n。当切片一个未知数量的维度时,建议传入 INT_MAX
。 axes
、 starts
和 ends
三个参数的元素数目必须相等。以下示例将解释切片如何工作:
- 示例1:
- 给定:
- data=[[1,2,3,4],[5,6,7,8],]
- axes=[0,1]
- starts=[1,0]
- ends=[2,3]
- 则:
- result=[[5,6,7],]
- 示例2:
- 给定:
- data=[[1,2,3,4],[5,6,7,8],]
- starts=[0,1]
- ends=[-1,1000] # 此处-1表示第0维的反向第0个位置,索引值是1。
- 则:
- result=[[2,3,4],] # 即 data[0:1, 1:4]
- 参数:
- input (Variable)- 多维
Tensor
或LoDTensor
,数据类型为float16
,float32
,float64
,int32
,或int64
。 - axes (list|tuple)- 数据类型是
int32
。表示进行切片的轴。它是可选的,如果不存在,将被视为 。 - starts (list|tuple|Variable)- 数据类型是
int32
。如果starts
的类型是 list 或 tuple,它的元素可以是整数或者形状为[1]的Tensor
或LoDTensor
。如果starts
的类型是Variable
,则是1-DTensor
或LoDTensor
。表示在各个轴上切片的起始索引值。 - ends (list|tuple|Variable)- 数据类型是
int32
。如果ends
的类型是 list 或 tuple,它的元素可以是整数或者形状为[1]的Tensor
或LoDTensor
。如果ends
的类型是Variable
,则是1-DTensor
或LoDTensor
。表示在各个轴上切片的结束索引值。
- input (Variable)- 多维
返回:多维 Tensor
或 LoDTensor
,数据类型与 input
相同。
返回类型:Variable。
- 抛出异常:
TypeError
:starts
的类型应该是 list、tuple 或 Variable。TypeError
:ends
的类型应该是 list、tuple 或 Variable。
代码示例:
- import paddle.fluid as fluid
- input = fluid.layers.data(
- name="input", shape=[3, 4, 5, 6], dtype='float32')
- # example 1:
- # attr starts is a list which doesn't contain tensor Variable.
- axes = [0, 1, 2]
- starts = [-3, 0, 2]
- ends = [3, 2, 4]
- sliced_1 = fluid.layers.slice(input, axes=axes, starts=starts, ends=ends)
- # sliced_1 is input[:, 0:3, 0:2, 2:4].
- # example 2:
- # attr starts is a list which contain tensor Variable.
- minus_3 = fluid.layers.fill_constant([1], "int32", -3)
- sliced_2 = fluid.layers.slice(input, axes=axes, starts=[minus_3, 0, 2], ends=ends)
- # sliced_2 is input[:, 0:3, 0:2, 2:4].