我有以下代表词向量的张量
A = (2, 500)
其中第一个维度是BATCH维度（即A包含两个词向量，每个词向量有500个元素）
我还有以下张量
B = (10, 500)
我想计算 A 和 B 之间的余弦距离，这样我得到
C = (2, 10, 1)
即对于 A 中的每一行计算与 B 中每一行的余弦距离
我查看了使用torch.nn.functional.F.cosine_similarity，但这不起作用，因为尺寸必须相同。
在 pytorch 中实现这一目标的最有效方法是什么？