import torch
a=torch.tensor([1,2])
b=torch.tensor([2,3])
a=set(a)
b=set(b)
print(a&b)
上述方式打印出为空集{},需要将tensor变量变成numpy变量
import torch
import numpy as np
a=np.array([1,2])
b=np.array([2,3])
a=set(a)
b=set(b)
print(a&b)
如果原本是tensor变量,则通过.numpy()将tensor转为numpy类型,这里不太清楚集合里面是tensor就无法正常使用一些集合操作的原因...