TensorFlow中的高階函數:tf.map_fn()

在TensorFlow中,有一些函數被稱爲高階函數(high-level function),和在python中的高階函數意義相似,其也是將函數當成參數傳入,以實現一些有趣的,有用的操作。其中tf.map_fn()就是其中一個。我們這裏介紹一下這個函數。 首先引入一個TF在應用上的問題:一般我們處理圖片的時候,常常用到卷積,也就是tf.nn.conv2d(),但是這個函數的輸入格式如:(batch
相關文章
相關標籤/搜索