lod_reset¶

paddle.fluid.layers. lod_reset ( x, y=None, target_lod=None ) [源代码] ¶

根据给定的参数 y 或 target_lod，重设输入 x (LoDTensor) 的 LoD 信息。

参数¶

x (Variable)：输入变量，类型为 Tensor 或者 LoDTensor。

y (Variable|None)：当 y 非空时，输出 LoDTensor 的 LoD 信息将与 y 的 LoD 一致。

target_lod (list|tuple|None)：一级 LoD，当 y 为空时，输出 LoDTensor 的 LoD 信息将与 target_lod 一致。

返回¶

Variable (LoDTensor)，重设了 LoD 信息的 LoDTensor。

返回类型¶

Variable (LoDTensor)。

抛出异常¶

TypeError：当 y 和 target_lod 二者均为空时抛出此异常。

          * 例 1:

x：包含一级 LoD 信息的 LoDTensor
    x.lod =  [[ 2,           3,                   1 ]]
    x.data = [[1.0], [2.0], [3.0], [4.0], [5.0], [6.0]]
    x.dims = [6, 1]

y: None

target_lod: [4, 2]

Output：包含一级 LoD 信息的 LoDTensor
    out.lod =  [[4,                          2]]
    out.data = [[1.0], [2.0], [3.0], [4.0], [5.0], [6.0]]
    out.dims = [6, 1]

* 例 2:

x：包含一级 LoD 信息的 LoDTensor
    x.lod =  [[2,            3,                   1]]
    x.data = [[1.0], [2.0], [3.0], [4.0], [5.0], [6.0]]
    x.dims = [6, 1]

y：普通 Tensor，不含 LoD 信息
    y.data = [[2, 4]]
    y.dims = [1, 3]

target_lod：当 y 不为空时，此参数不起作用

Output：包含一级 LoD 信息的 LoDTensor
    out.lod =  [[2,            4]]
    out.data = [[1.0], [2.0], [3.0], [4.0], [5.0], [6.0]]
    out.dims = [6, 1]

* 例 3:

x：包含一级 LoD 信息的 LoDTensor
    x.lod =  [[2,            3,                   1]]
    x.data = [[1.0], [2.0], [3.0], [4.0], [5.0], [6.0]]
    x.dims = [6, 1]

y：包含二级 LoD 信息的 LoDTensor
    y.lod =  [[2, 2], [2, 2, 1, 1]]
    y.data = [[1.1], [2.1], [3.1], [4.1], [5.1], [6.1]]
    y.dims = [6, 1]

target_lod：当 y 不为空时，此参数不起作用

Output：包含二级 LoD 信息的 LoDTensor
    out.lod =  [[2, 2], [2, 2, 1, 1]]
    out.data = [[1.0], [2.0], [3.0], [4.0], [5.0], [6.0]]
    out.dims = [6, 1]

         

代码示例¶

          import paddle.fluid as fluid
x = fluid.layers.data(name='x', shape=[10])
y = fluid.layers.data(name='y', shape=[10, 20], lod_level=2)
out = fluid.layers.lod_reset(x=x, y=y)

         