Вы можете попробовать это.Здесь placeholder[:1]
- синтаксис среза для получения первого значения, которое 1
u = u.assign(placeholder[:1])
- это присвоение первого элемента.
import tensorflow as tf
placeholder = tf.placeholder( tf.int32, shape=(4,))
def Graph(placeholder):
# I want to store the first value of placeholder in "u"
with tf.variable_scope("reuse", reuse=tf.AUTO_REUSE):
u = tf.get_variable('x',[1],dtype=tf.int32, trainable=False)
u = u.assign(placeholder[:1])
place_print = tf.Print(placeholder[:1],[placeholder[:1]])
u_print = tf.Print(u,[u])
# Some calculation including 'u'
y = 7 * place_print - u_print
return y
with tf.Session() as sess:
with tf.variable_scope("reuse", reuse=tf.AUTO_REUSE):
u = tf.get_variable("x", [1], dtype=tf.int32)
f = Graph(placeholder)
sess.run( tf.global_variables_initializer() )
print(sess.run( [f,u],feed_dict={placeholder: [1, 2, 3, 4]}))
Выходные данные
[array([6]), array([0])]