merge_selected_rows

paddle.fluid.layers. merge_selected_rows ( x, name=None ) [源代码]

累加合并 SelectedRows ( x ) 中的重复行，并对行值由小到大重新排序。

参数:

  • x (Variable) : 类型为 SelectedRows，选中行允许重复。

  • name (basestring|None) : 输出变量名称。

返回:

  • 含有 SelectedRows 的 Variable，选中行不重复。

返回类型:

  • Variable（变量）。

代码示例

  1. import paddle.fluid as fluid
  2. import numpy
  4. place = fluid.CPUPlace()
  5. block = fluid.default_main_program().global_block()
  7. var = block.create_var(name="X2",
  8.                        dtype="float32",
  9.                        persistable=True,
  10.                        type=fluid.core.VarDesc.VarType.SELECTED_ROWS)
  11. y = fluid.layers.merge_selected_rows(var)
  12. z = fluid.layers.get_tensor_from_selected_rows(y)
  14. x_rows = [0, 2, 2, 4, 19]
  15. row_numel = 2
  16. np_array = numpy.ones((len(x_rows), row_numel)).astype("float32")
  18. x = fluid.global_scope().var("X2").get_selected_rows()
  19. x.set_rows(x_rows)
  20. x.set_height(20)
  21. x_tensor = x.get_tensor()
  22. x_tensor.set(np_array, place)
  24. exe = fluid.Executor(place=place)
  25. result = exe.run(fluid.default_main_program(), fetch_list=[z])
  27. print("x_rows: ", x_rows)
  28. print("np_array: ", np_array)
  29. print("result: ", result)
  30. '''
  31. Output Values:
  32. ('x_rows: ', [0, 2, 2, 4, 19])
  33. ('np_array: ', array([[1., 1.],
  34.        [1., 1.],
  35.        [1., 1.],
  36.        [1., 1.],
  37.        [1., 1.]], dtype=float32))
  38. ('result: ', [array([[1., 1.],
  39.        [2., 2.],
  40.        [1., 1.],
  41.        [1., 1.]], dtype=float32)])
  42. '''