标签: tensorflow keras
我大约有500k行的512维int16值,这些值将在训练期间被检索并用于模型输入,并基于索引进行检索。
我看到我可以使用tf.constant存储数据,并使用tf.gather查找值,也可以使用tf.lookup.StaticHashTable来存储/检索值。
tf.constant
tf.gather
tf.lookup.StaticHashTable
做一个比做另一个有优势吗?
如果有帮助,我将通过TPU训练我的模型。