与使用tf.gather和tf.constant相比,使用tf.lookup.StaticHashTable有什么优势?

时间:2020-07-20 06:03:25

标签: tensorflow keras

我大约有500k行的512维int16值,这些值将在训练期间被检索并用于模型输入,并基于索引进行检索。

我看到我可以使用tf.constant存储数据,并使用tf.gather查找值,也可以使用tf.lookup.StaticHashTable来存储/检索值。

做一个比做另一个有优势吗?

如果有帮助,我将通过TPU训练我的模型。

0 个答案:

没有答案