2024-04-25 22:58:08 发布
网友
给定一个具有唯一键和值的固定字典(所有数字都是非负整数):
d = {(1,3): 6, (5,4): 9}
在tensorflow中,什么是转换的有效方法:
反之亦然:
tf.constant([6, 9, 6]) -> tf.constant([[1,3], [5,4], [1,3]])
tensorflow中有一个可查找的模块-文档位于: https://www.tensorflow.org/api_docs/python/tf/contrib/lookup
正如@OphirYoktan提到的,有一个查找操作符。我建议使用^{},但由于您的目标是同时映射vector - id,因此可以执行以下操作。在
vector - id
使用^{}
d_inverse = {v:k for k,v in d.items()} d_mapped = tf.map_fn(lambda x: d[x], d.values()) d_mapped_inverse = tf.map_fn(lambda x: d_inverse[x], d_inverse.items()}
您只需将dicts d和d_inverse中的值设置为tf.constants
d
d_inverse
tf.constants
话虽如此,这些映射应该与关于图的计算原因有关,否则我建议您在图之外进行映射。在
tensorflow中有一个可查找的模块-文档位于: https://www.tensorflow.org/api_docs/python/tf/contrib/lookup
正如@OphirYoktan提到的,有一个查找操作符。我建议使用^{} ,但由于您的目标是同时映射
vector - id
,因此可以执行以下操作。在使用^{}
您只需将dicts
d
和d_inverse
中的值设置为tf.constants
话虽如此,这些映射应该与关于图的计算原因有关,否则我建议您在图之外进行映射。在
相关问题 更多 >
编程相关推荐