函数中的张量流占位符

Question

现在，我有一些如下代码：

import tensorflow as tf

x = tf.placeholder(tf.float32, [1, 3])
y = x * 2

with tf.Session() as sess:
    result = sess.run(y, feed_dict={x: [2, 4, 6]})
    print(result)

我想为函数提供值。

下面是一个明显的错误。 我该怎么做呢？

import tensorflow as tf

def place_func():
    x = tf.placeholder(tf.float32, [1, 3])
    y = x * 2

with tf.Session() as sess:
    result = sess.run(y, feed_dict={x: [2, 4, 6]})
    print(result)

Answer 1

一种选择是在函数内部运行会话，如下所示：

import tensorflow as tf

def my_func(data):
    x = tf.placeholder(tf.float32, [1, 3])
    y = x * 2
    return sess.run(y, feed_dict = {x: data})

with tf.Session() as sess:
    result = my_func([[2, 4, 6]])
    print(result)

您也可以创建一个类，并让 x 成为该类的一个字段。