tf.py_func函数用来扩展tensorflow的灵活性

tf.py_func(func, inp, Tout)

在这里第一个参数是核心,也就是一个用户自定义的函数,输入是numpy array,输出也是numpy array,在该函数中,用户可以自由的使用numpy 操作

第二个参数是inp,是func函数接受的输入,是一个列表。

第三个参数是Tout,指定了func函数返回后的numpy array 转化成的tesnor后的格式,如果是多个值就是列表否则就是一个元组,如果是一个返回值,就是一个单独的dtype的类型

ty.py_func的输出:

输出是一个tensor列表,或者是一个单个的tensor。

具体详情请参照链接:

https://blog.csdn.net/jiongnima/article/details/80555387

猜你喜欢

转载自blog.csdn.net/weixin_41950276/article/details/83590058
今日推荐