1. 程式人生 > >tf.py_func函式用來擴充套件tensorflow的靈活性

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