如何在张量流中对此进行矢量化?

时间:2018-02-28 07:37:15

标签: tensorflow vectorization

让A天是M×J矩阵,B是N×J矩阵。

我想生成一个M×N矩阵S,其中:

S_ij = w^t[A_i;B_j;A_i o B_j)  

基本上,结果矩阵S的每个元素是一些向量w(矩阵W的一行)点积,其中:col A_j与col B_j连接,并且与所有i和每个A_i和B_j的元素乘积相连。学家

理想情况下,我想对操作进行矢量化,并且只使用矩阵形式S,A和B而不是切片和执行循环。然而,我是tensorflow的新手,并且似乎无法弄清楚如何以矩阵形式执行代码,因为A_i o B_j似乎导致2d矩阵,并且它需要与向量A_i和B_j连接。

由于

0 个答案:

没有答案