假设我有一个矩阵 A
和一个矩阵 B
. 我知道 tf.matmul(A,B)
可以计算两个矩阵的乘法 . 但是我有一项任务只需要将 A
的某些行与某些 B
列相乘 .
例如,我有一个 A
, Ls_A=[0,1,2]
的行ID列表,以及 B
, Ls_B=[4,2,6]
的列ID列表 . 我想要一个列表的结果,表示为 Ls
,这样:
Ls[0] = A[0,:] * B[:,4]
Ls[1] = A[1,:] * B[:,2]
Ls[2] = A[2,:] * B[:,6]
我怎样才能做到这一点?
谢谢大家的帮助!
1 回答
您可以使用tf.gather执行以下操作:
这打印