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

Tensorflow 如何有效地用零替换秩2张量的最后一行?,tensorflow,Tensorflow,假设我有一个秩为2的张量(矩阵)。我想用零填充这个预先存在的矩阵的最后一行。我不希望tensorflow在一个新的地方复制整个矩阵,因为它是巨大的。有可能吗?答案基于David Parks的建议,即研究以下主题: 根据这个答案,我得出了问题的确切解决方案: a = tf.Variable(tf.ones([10, 36, 36])) value = tf.zeros([36, 36]) d = tf.scatter_update(a, 9 , value) with tf.Ses

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

答案基于David Parks的建议,即研究以下主题:

根据这个答案,我得出了问题的确切解决方案:

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)

这回答了你的问题吗?