如何在每行的不同位置合并一对2d张量

时间:2019-03-28 21:38:24

标签: tensorflow

所需行为:

t1 = [[1, 2, 0], [3, 0, 0]]
t2 = [[4, 0, 0], [5, 6, 1]]
t3 = tf.magical_op(t1, t2)

结果就是

t3 = [[1, 2, 4], [3, 5, 6]]

换句话说,我需要用t2中的元素替换t1中的0。 将t1行中的第一个0与t2同一行中的第一个元素进行匹配, 对于第二个0为t2的第二个元素,依此类推。

0 个答案:

没有答案