是否可以将tf.SparseTensor
与tf.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()