gpt4 book ai didi

python - 如何检查不同张量pytorch中的张量值?

转载 作者:行者123 更新时间:2023-12-04 08:09:52 24 4
gpt4 key购买 nike

我有 2 个大小不等的张量

a = torch.tensor([[1,2], [2,3],[3,4]])
b = torch.tensor([[4,5],[2,3]])

我想要一个 bool 数组,表示每个值是否存在于另一个张量中而不进行迭代。类似的东西

a in b

结果应该是

[False, True, False]

因为 b 中只有 a[1] 的值

最佳答案

我认为如果不使用至少某种类型的迭代,这是不可能的。我可以管理的最简洁的方法是使用列表理解:

[i in b for i in a]

检查 b 中是否存在 a 中的元素并给出 [False, True, False]。也可以反过来得到b中的元素a[False, True]。

关于python - 如何检查不同张量pytorch中的张量值?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/66036375/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com