可以用集合运行实现,但转换为set之后会删除重复值,这里借用np.isin实现。
A = torch.tensor([1, 2, 3]) B = torch.tensor([2]) A[~np.isin(A, B)]
np.isin还可以用于dataframe数据的过滤中,速度很快非常好用。
如果有现成的api或者更好的方法,麻烦告知,十分愿意学习。

可以用集合运行实现,但转换为set之后会删除重复值,这里借用np.isin实现。
A = torch.tensor([1, 2, 3]) B = torch.tensor([2]) A[~np.isin(A, B)]
np.isin还可以用于dataframe数据的过滤中,速度很快非常好用。
如果有现成的api或者更好的方法,麻烦告知,十分愿意学习。