使用tf.boolean_mask时获取UserWarning

Question

我在使用tf.boolean_mask()时获得了此UserWarning：

/usr/local/lib/python2.7/site-packages/tensorflow/python/ops/gradients_impl.py:93: UserWarning: Converting sparse IndexedSlices to a dense Tensor of unknown shape. This may consume a large amount of memory.
  "Converting sparse IndexedSlices to a dense Tensor of unknown shape. "

删除后，警告信息消失。我找到了this answer，但我没有使用tf.gather()，我确信我已将tf.Variable和tf.placeholder传递给tf.nn.embedding_lookup()。那么我怎么能解决这个问题呢？

感谢您的帮助！

Answer 1

同样，由于tf.boolean_mask（）导致此错误。查看图表，您可以看到boolean_mask函数实际上包含对tf.gather（）的调用。

根据AI_ROBOT在the answer you mentioned中提出的建议，即使用dynamic_partition（）而不是boolean_mask（），为我做了诀窍！