tf.gather, tf.gather_nd和tf.slice

在tensorflow和pytorch中,都有一些函數是用於對張量進行索引的,因爲pytorch是基於動態圖的,它的張量可以動態的進行操作,不需要在對話(session)中進行,因此可能採用這些方法的必要性沒有那麼足,但是因爲tensorflow是基於靜態圖的,因此如果在構建計算圖的過程中需要對張量進行索引,一般是不能進行簡單的slice的,因此引入這些函數就很有必要了,我們這裏集中介紹下這些函數
相關文章
相關標籤/搜索