我有一个这样的 Torch Tensor:
a=[1, 234, 54, 6543, 55, 776]
字符串
和其他Tensor一样:
b=[234, 54]
c=[55, 776]
型
我想创建一个新的掩码Tensor,如果有另一个Tensor(b
或c
)与a
相等,则a
的值将为true。
例如,在上面的Tensor中,我想创建以下掩码Tensor:
a_masked =[False, True, True, False, True, True]
# The first two True values correspond to tensor `b` while the last two True values
correspond to tensor `c`.
型
我见过其他方法来检查一个完整的Tensor是否包含在另一个Tensor中,但这里不是这样。
有没有一个 Torch 的方式来做到这一点有效?谢谢!
1条答案
按热度按时间cgyqldqp1#
根据PyTorch论坛here上的答案,您可以显式地使用for循环,例如,
字符串
然而,实际上有一个PyTorch
isin
函数,你可以这样做:型
这比
sum
方法快几倍。