Quantcast
Channel: How to feed a local placeholder of some function from main function? - Stack Overflow
Viewing all articles
Browse latest Browse all 2

Answer by pfm for How to feed a local placeholder of some function from main function?

$
0
0

If you want to access a placeholder or a tensor directly you can call get_tensor_by_name:

import tensorflow as tf

def f():
    a=tf.placeholder(dtype=tf.float32, name='name_a')
    b=tf.placeholder(dtype=tf.float32, name='name_b')
    addition=tf.add(a,b)
    return addition

if __name__ == '__main__':
    var = f()
    a = tf.get_default_graph().get_tensor_by_name('name_a:0')
    b = tf.get_default_graph().get_tensor_by_name('name_b:0')
    with tf.Session() as sess:
        print(sess.run(var, feed_dict={a:[1],b:[2]}))

[3.]


Viewing all articles
Browse latest Browse all 2

Trending Articles



<script src="https://jsc.adskeeper.com/r/s/rssing.com.1596347.js" async> </script>