张量流-将两个向量相乘以生成矩阵

Question

我有两个向量，例如row = [1,2,3,4] col = [0,1,2,3]，我想将它们相乘以产生一个矩阵：

[
  [0,0,0,0],
  [1,2,3,4],
  [2,4,6,8],
  [3,6,9,12]
]

我该如何在tensorflow中做到这一点？

Answer 1

首先，您必须将它们转换为Tensorflow tf.constants，然后可以使用tf.matmul将它们相乘：

a = tf.constant(row, shape=[4, 1])
b = tf.constant(col, shape=[1, 4])

result = tf.matmul(row,col)