如何用零有效地替换rank-2张量的最后一行?

时间:2017-05-15 22:42:08

标签: tensorflow

让我们说我有一个秩-2张量(一个矩阵)。我想用零填充这个预先存在的矩阵的最后一行。我不希望tensorflow在一个新的地方复制整个矩阵,因为它很大。有可能吗?

1 个答案:

答案 0 :(得分:0)

答案是基于David Parks关于调查此主题的建议: How to do slice assignment in Tensorflow

使用这个答案,我已经找到了问题的确切解决方案:

a = tf.Variable(tf.ones([10, 36, 36]))   
value = tf.zeros([36, 36])   
d = tf.scatter_update(a, 9 , value)

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print a.eval(session=sess)
    sess.run(d)
    print a.eval(session=sess)