pytorch 一个tensor去除指定值的元素

可以用集合运行实现，但转换为set之后会删除重复值，这里借用np.isin实现。

A = torch.tensor([1, 2, 3])
B = torch.tensor([2])

A[~np.isin(A, B)]

np.isin还可以用于dataframe数据的过滤中，速度很快非常好用。

如果有现成的api或者更好的方法，麻烦告知，十分愿意学习。