我试图在Tensorflow(而不是原生Python或NumPy或pandas)中实现一些非常简单的事情,可以通过以下任何方式完成:
有2个不同大小的独立阵列/张量。每个元素都包含两个值:比较值和权重。我们想要比较两个张量中的比较值,并乘以它们相应的权重。
将比较值和权重作为不同的数组。然后比较比较值,得到指数,然后使用指数在权重向量中找到元素,然后将它们相乘。
简而言之,我想在张量中找到匹配元素的索引。
我能找到的最接近的解决方案是将它们转换为集合,但它没有给出元素的确切索引。
我能够用Pandas实现我想要的目标:
matched = pd.Index(v1).intersection(pd.Index(v2))
和原生Python:
ind_v1, ind_v2 = [i for i, item in enumerate(v1_1) if item in v2_1],[i for i, item in enumerate(v2_1) if item in v1_1]
我希望在Tensorflow中也能这样做。