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