从SparseTensorValue访问值

时间:2018-08-31 07:35:02

标签: python tensorflow tfrecord

我正在解析一个tfrecord文件。我在SparceTensorValue中得到一些坐标。这是我从印刷品中得到的:

>> print("coords: ", coords)

>> coords:  SparseTensorValue(indices=array([[0],
       [1],
       [2],
       [3],
       [4]]), values=array([0.19      , 0.43666667, 0.555     , 0.7416667 , 0.90166664],
      dtype=float32), shape=array([5]))

如何整齐地访问值?

我想要的是一个numpy数组:

[0.19, 0.44, 0.55, 0.74, 0.90]

1 个答案:

答案 0 :(得分:0)

这实际上比我想象的要容易。原来我要做的只是以下内容:

>> print("coords: ",coords.values)
>> coords:  [0.19       0.43666667 0.555      0.7416667  0.90166664]