如何进行矩阵相等？
How to perform matrix equality?
我有两个矩阵。 A 尺寸为 160 x 250，B 尺寸为 3200 x 250。
我想获取 A 的每一行与 B 的每一行的集合交集，以获得 160 x 3200 的向量。 (集合大小为250个元素)
有什么实现方法吗？
我认为它应该需要 torch.eq，但不确定如何更改尺寸。例如：
result = torch.sum(torch.eq(A[0], B), dim=1) 会给我一个 3200 元素向量与 A 的第 0 行的比较。我想要A 的所有行 (160)
假设您要检查 160x3200 可能的 250 个特征向量对之间的相等性。您可以使用索引技巧来做到这一点：
>>> (A[None] == B[:, None]).all(-1)
我有两个矩阵。 A 尺寸为 160 x 250，B 尺寸为 3200 x 250。
我想获取 A 的每一行与 B 的每一行的集合交集，以获得 160 x 3200 的向量。 (集合大小为250个元素)
有什么实现方法吗？
我认为它应该需要 torch.eq，但不确定如何更改尺寸。例如：
result = torch.sum(torch.eq(A[0], B), dim=1) 会给我一个 3200 元素向量与 A 的第 0 行的比较。我想要A 的所有行 (160)
假设您要检查 160x3200 可能的 250 个特征向量对之间的相等性。您可以使用索引技巧来做到这一点：
>>> (A[None] == B[:, None]).all(-1)