Python Pytorch批量矩阵外积

类似于中的问题，我有两个矩阵，想计算它们的外积，或者换句话说，成对元素的积

形状示例： 如果我们有X1和X2的形状

torch.Size（[32300,8]）

---

结果应为大小

torch.size（[323003008]）

您可以添加单个维度：

X1[：，无，…]*X1[…，无，：]

但乌斯曼·阿里也是个好主意。使用：

torch.einsum（'bik，bjk->bijk'，X1，X2）

您可以添加单个维度：

X1[：，无，…]*X1[…，无，：]

但乌斯曼·阿里也是个好主意。使用：

torch.einsum（'bik，bjk->bijk'，X1，X2）

torch.einsum（'bik，bjk->bijk'，X1，X2）？torch.einsum（'bik，bjk->bijk'，X1，X2）怎么样？