tensorflow编程: Wraps python functions
tensorflow编程: Wraps python functions
JNingWei
发布于 2018-09-28 15:23:34
发布于 2018-09-28 15:23:34
Script Language Operators
tf.py_func
在 tensorflow 中 缺乏 需要的 函数接口 时，负责将任意的 python/numpy functions 包装成 TensorFlow op。
tf.py_func (func, inp, Tout, stateful=True, name=None)
其中， inp参数项 必须是 list型 ，哪怕 list 里只有一个元素；而 inp参数项 在只有一个元素时可以 不必为 list型 。
示例一：
import tensorflow as tf
import numpy as np
inputs = 1.
my_func = lambda x : np.sinh(x)
y = tf.py_func(my_func, [inputs], tf.float32)
sess = tf.InteractiveSession()
print y.eval()1.1752示例二：
import tensorflow as tf
import numpy as np
array1 = np.array([[1, 2], [3, 4]], dtype=np.float32)
array2 = np.array([[5, 6], [7, 8]], dtype=np.float32)
def add_minus_dot(array1, array2):
return array1 + array2, array1 - array2, np.dot(array1, array2)
y1, y2, y3 = tf.py_func(add_minus_dot, [array1, array2], [tf.float32, tf.float32, tf.float32])
with tf.Session() as sess:
_y1, _y2, _y3 = sess.run([y1, y2, y3])
print _y1
print
print _y2
print
print _y3[[ 6. 8.]
[ 10. 12.]]
[[-4. -4.]
[-4. -4.]]
[[ 19. 22.]
[ 43. 50.]]本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2017年10月05日，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录