如果输入被tf.data.TFRecordDataset随机混编，Tensorflow如何获取输入和输出之间的映射关系

Question

给定输入数据b，user_vector的每一行都是一个用户的向量，用户是唯一的。通过tf.data.TFRecordDataset读取输入数据，并随机对其进行随机排序，现在我们得到user_vector。假设shuffled_user_vector经过神经网络并变成shuffled_user_vector。我的问题是如何将user_embedding映射到user_vector，即user_embedding和user_vector中的哪些行指向同一用户？

我应该保留随机用户向量的索引吗？还是有更好的张量流解决方案？

Answer 1

您可以将user-id属性添加到数据管道中（在重排之前），并且在进行预测时，将此属性与user_embedding一起返回。