两批之间的成对相似度计算无需迭代？

Question

我有两个3D阵列。第一个的大小为1000x1x5，第二个的大小为1000x10x5，其中1000是批处理大小。

我想计算1x5行向量与其批次B中对应的10个行向量之间的成对相似度。

可以使用sklearn https://scikit-learn.org/stable/modules/classes.html#module-sklearn.metrics.pairwise通过在批处理中进行迭代来完成。我的问题是是否可以不进行迭代？