get_tensor_from_selected_rows

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

该OP从SelectedRows类型的输入中获取向量数据，以LoDTensor的形式输出。

  1. 例如：
  2.  
  3.       输入为SelectedRows类型:
  4.            x.rows = [0, 5, 5, 4, 19]
  5.            x.height = 20
  6.            x.value = [[1, 1] [2, 2] [2, 2] [3, 3] [6, 6]]
  7.  
  8.       输出为LoDTensor：
  9.            out.shape = [5, 2]
  10.            out.data = [[1, 1],
  11.                        [2, 2],
  12.                        [2, 2],
  13.                        [3, 3],
  14.                        [6, 6]]
  • 参数：
    • x (SelectedRows) - SelectedRows类型的输入，数据类型为float32，float64，int32或int64。
    • name (str) - 该参数供开发人员打印调试信息时使用，具体用法请参见 Name ，默认值为None。

返回： 从SelectedRows中转化而来的LoDTensor，数据类型和输入一致。

返回类型： Variable

代码示例：

  1. import paddle.fluid as fluid
  2. b = fluid.default_main_program().global_block()
  3. input = b.create_var(name="X", dtype="float32", persistable=True, type=fluid.core.VarDesc.VarType.SELECTED_ROWS)
  4. out = fluid.layers.get_tensor_from_selected_rows(input)