如何使用tensoflow进行这样的计算

Question

两个张量 x = [a，b，c] y = [d，e，f] 两个float32号码 我想得到像这样的张量z = [ad，ae，af，bd，be，bf，cd，ce，cf] 这是一个简单的例子 真正的张量可能是巨大的

Answer 1

我想你可以将x重塑为形状（3,1），y形状（1,3）。然后当你将它们相乘时，它们将广播并给你一个可以展平的（3,3）数组。这是一个numpy的例子。 Tensorflow以完全相同的方式运行并具有相同的功能（除了使用tf.reshape(z, [-1])进行展平，因为张量流中没有展平。）

x = np.array([1, 2, 3])
y = np.array([4, 5, 6])
z = np.expand_dims(x, 1) * np.expand_dims(y, 0)
z = z.flatten()

这给了我array([ 4, 5, 6, 8, 10, 12, 12, 15, 18])作为输出。