1
我有两个向量X = [a,b,c,d]
和Y = [m,n,o]
。我想从X和Y即如何从张量流中两个向量的元素组合创建矩阵
M[j,i] = f(X[i], Y[j])
# e.g. where f(x,y) = x-y:
M :=
a-m b-m c-m d-m
a-n b-n c-n d-n
a-o b-o c-o d-o
我想我可以具有两个tf.while_loop()执行此构造的矩阵M,其中每个元素是在每对的操作,但是,似乎效率低下,我想知道是否有更紧凑和平行的方式来做到这一点。
P.S. X和Y实际上并不是矢量,而是R2。即X和Y中的每个元素本身就是一个固定长度的向量,并且f(X,Y)元素地执行f()。另外还有一个批处理组件。
I.e.
X.shape => [BATCH, I, K]
Y.shape => [BATCH, J, K]
M[batch, j, i, k] = f(X[batch, i, k], Y[batch, j, k])
# e.g.:
= X[batch, i, k] - Y[batch, j, k]
这是使用Python API BTW