所有成对点积pytorch

问题描述

是否有一个内置函数可以有效地计算 Pytorch 中两个张量的所有成对点积？ 例如
输入 - 张量 A（形状 NxD）
张量 B（形状 NxD）

输出 - 张量 C（形状 NxN）使得 C_i,j = torch.dot(A_i,B_j) ?

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）