Created by: aprilvkuo

demo代码

  def forward(self, x, y, split_list):
        """
        x -> batch_size, n
        y ->  1, n
       split_list 是一个variable， 通过feed传进去数据
        """
        split_list = np.array(split_list)
        print(type(split_list))
        print(split_list)

从输出结果也看出来， 可以通过np.array(*) 方式转化，但是不是标准的numpy的数据格式，没法获取其中的数据。

输出结果：

<type 'numpy.ndarray'>
name: "split_list"
type {
  type: LOD_TENSOR
  lod_tensor {
    tensor {
      data_type: INT64
      dims: 3
    }
    lod_level: 0
  }
}
persistable: false