两个tensor怎么计算交集

如果两个tensor是一维的，可以使用intersect1d函数来计算它们的交集。例如：

import torch

a = torch.tensor([1, 2, 3, 4, 5])
b = torch.tensor([3, 4, 5, 6, 7])
intersect = torch.intersect1d(a, b)
print(intersect)

输出结果为：

tensor([3, 4, 5])

如果两个tensor是多维的，可以使用reshape和view函数将它们转换为一维的，然后再使用intersect1d函数计算它们的交集。例如：

import torch

a = torch.tensor([[1, 2], [3, 4], [5, 6]])
b = torch.tensor([[3, 4], [5, 6], [7, 8]])
intersect = torch.intersect1d(a.reshape(-1), b.reshape(-1))
intersect = intersect.view(-1, 2)
print(intersect)

输出结果为：

tensor([[3, 4],
        [5, 6]])

这样就得到了两个多维tensor的交集。