tf.py_func(func, inp, Tout)
在这里第一个参数是核心,也就是一个用户自定义的函数,输入是numpy array,输出也是numpy array,在该函数中,用户可以自由的使用numpy 操作
第二个参数是inp,是func函数接受的输入,是一个列表。
第三个参数是Tout,指定了func函数返回后的numpy array 转化成的tesnor后的格式,如果是多个值就是列表否则就是一个元组,如果是一个返回值,就是一个单独的dtype的类型
ty.py_func的输出:
输出是一个tensor列表,或者是一个单个的tensor。
具体详情请参照链接: