如何将不规则张量与 tf.data 和 TFRecords 一起使用?

zer*_*ool 5 python tensorflow tensorflow-datasets

Tensorflow 最近发布了 Ragged Tensors:https://www.tensorflow.org/guide/ragged_tensors

但没有任何文档说明如何将不规则数据保存为 TFRecord,并使用数据 api 恢复它。

小智 1

不幸的是,没有RaggedFeature或同等功能。您最好的选择可能是转换为稀疏(通过to_sparse())并将数据编码为SparseFeature。解码后,您可以通过构建器将其转换回不规则状态from_sparse()