使用IndexedSlices调整tensorflow

时间:2018-10-22 20:54:41

标签: tensorflow

我正在使用tensorflow对IR任务执行lambdaRank。我使用了嵌入(tf.gather),因此它们会将IndexedSlices作为grads返回(我们称它们为grad_emb)。我必须调整这些IndexedSlices,即将它们乘以一个密集的张量lambda。

total_grad_emb = lambda (dense) * grad_emb (IndexedSlices)

我试图将grad_emb(稀疏格式的indexsslice)转换为密集张量,但它报告了OOM,因此对于grad_emb部分,我必须在稀疏模式下进行所有操作。现在出现了一个问题,即如何将lambda(密集)转换为IndexedSlices,以便它们可以相乘。谢谢

0 个答案:

没有答案