是否可以将tf.SparseTensor与tf.data.Dataset一起使用?

时间:2018-11-13 13:03:01

标签: python tensorflow tensorflow-datasets

是否可以将tf.SparseTensortf.data.Dataset一起使用?

我正在尝试保留一个大型的稀疏矩阵,但没有找到任何示例来实现此目的,但是在1.5.0版中它表示支持。

这是我今天生成数据集的方式,但是它生成非常大的文件:

features = tf.train.Features(feature={
        'label': tf.train.Feature(int64_list=tf.train.Int64List(value=[label])),
        'features': tf.train.Feature(float_list=tf.train.FloatList(value=dense_array)) })
    example = tf.train.Example(features=features)

writer = tf.python_io.TFRecordWriter("filename.tfreacords") writer.write(example) write.close()

0 个答案:

没有答案