寻找两个pytorch张量的不交

import torch

# 创建两个张量
tensor1 = torch.tensor([1, 2, 3, 4])
tensor2 = torch.tensor([3, 4, 5, 6])

# 将两个张量按维度0拼接
concat_tensor = torch.cat((tensor1, tensor2), dim=0)

# 找出拼接后张量中的唯一值
unique_values = torch.unique(concat_tensor)

# 将唯一值与原来的两个张量进行比较，得到不交集元素
not_intersect = torch.eq(unique_values.unsqueeze(1), tensor1.unsqueeze(0)).sum(dim=1) + \
               torch.eq(unique_values.unsqueeze(1), tensor2.unsqueeze(0)).sum(dim=1) == 1
intersect_tensor = unique_values[not_intersect]

print("不交集元素：", intersect_tensor)