如何在2个不同尺寸的张量中找到匹配元素?

时间:2017-09-22 15:08:39

标签: tensorflow compare

我试图在Tensorflow(而不是原生Python或NumPy或pandas)中实现一些非常简单的事情,可以通过以下任何方式完成:

  1. 有2个不同大小的独立阵列/张量。每个元素都包含两个值:比较值和权重。我们想要比较两个张量中的比较值,并乘以它们相应的权重。

  2. 将比较值和权重作为不同的数组。然后比较比较值,得到指数,然后使用指数在权重向量中找到元素,然后将它们相乘。

  3. 简而言之,我想在张量中找到匹配元素的索引。

    我能找到的最接近的解决方案是将它们转换为集合,但它没有给出元素的确切索引。

    我能够用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中也能这样做。

0 个答案:

没有答案