如何在 TensorFlow 中实现这个操作？

【问题标题】：How to implement this operation in TensorFlow?如何在 TensorFlow 中实现这个操作？
【发布时间】：2020-04-19 21:09:05
【问题描述】：

假设我们有两个形状的张量 X,Y，分别是 (a,b) 和 (b,c)。然后我们可以构造一个由T[i][j][k] = X[i][j] + Y[j][k] 定义的形状为 (a,b,c) 的新张量 T。

如何在 TensorFlow 中实现这一点？我用乘法代替加法的模拟怎么样？

提前谢谢你。

【问题讨论】：

标签： tensorflow tensor

【解决方案1】：

这就是你要找的吗？

a = tf.constant([[2,3],[4,1]]) # A 2x2 matrix
b = tf.constant([[1,2,1],[3,4,2]]) # A 2x3 matrix

# Make a into 2x2x3 matrix
aa = tf.expand_dims(a,-1)
# Make b into 1x2x3 matrix
bb = tf.expand_dims(b, 0)

# Do element wise addition, resulting in 2x2x3 matrix.
c = aa + bb

【讨论】：

是的！非常感谢。一个问题：你为什么要重复一个？广播不处理吗？
@user11267981 是的，实际上应该没问题。我当时没有意识到。