Python 如何更新匹配维度向量的张量

设有两个二维张量，A（

m×c

）和B（

n×c

）。属于B的每一行向量也属于A即。此外，A中的行向量不是唯一的，即A可能有重复的行。但是，B中的行向量是唯一的

还有另一对张量p（

m×f

）和Q（

n×f

）。我正在努力做到以下几点

范围内i的

（B.shape[0]）：
rv=B[i，：]
fv=Q[i，：]
#P[，：]=fv

如何正确地做到这一点

有可能去掉for循环吗

---

您可以使用以下掩码：

范围内i的

（B.shape[0]）：
rv=B[i]
fv=Q[i]
面罩=火炬。其中（（A==rv）。全部（尺寸=1））[0]
P[mask]=fv

谢谢。有什么办法可以完全消除for循环？@tachyon因为你需要在

a

中查找

B

中的每一行，我想不出任何方法可以避免引擎盖下的for循环。