僅將張量添加到另一個張量的一部分

Question

我必須添加兩個張量，一個張量在深度方向上是另一個張量的形狀。 這是一個例子

t1 = tf.constant(3, shape=[2, 2, 2], dtype=tf.float32)
t2 = tf.constant(1, shape=[2, 2, 1], dtype=tf.float32)

我想使用tf.add東西將第二張量添加到第一個張量，但僅在形狀的第三部分的第一層中添加。 帶數字

t1 = [[[3, 3], [3, 3]],
      [[3, 3], [3, 3]]]
t2 = [[[1, 1], [1, 1]]]

output = [[[4, 4], [4, 4]],
          [[3, 3], [3, 3]]]

有內置的功能可以做到這一點嗎？

Answer 1

將t1的第一個“ 列 ”與t2 ，然后將其與t1的其余列合並：

t1 = tf.constant(3, shape=[2, 2, 2], dtype=tf.float32)
t2 = tf.constant(1, shape=[2, 2, 1], dtype=tf.float32)
tf.InteractiveSession()

tf.concat((t1[...,0:1] + t2, t1[...,1:]), axis=2).eval()

#array([[[4., 3.],
#        [4., 3.]],

#       [[4., 3.],
#        [4., 3.]]], dtype=float32)

請注意，您的第二個示例t2具有不同的形狀，即(1,2,2)而不是(2,2,1) ，在這種情況下，第一個軸為slice和concat：

tf.concat((t1[0:1] + t2, t1[1:]), axis=0).eval()

#array([[[4., 4.],
#        [4., 4.]],

#       [[3., 3.],
#        [3., 3.]]], dtype=float32)