1. 程式人生 > >tensorflow 中tf.expand_dims函式用法

tensorflow 中tf.expand_dims函式用法

tf.expand_dims()(維度擴充套件)函式用法

tf.expand_dims(input,position,name)    

其中input就是要擴充套件的變數,position就是選擇在那個位置上擴充套件維度,如果input原來是3維的,那麼position有四個數字(0,1,2,3)可以選擇可以理解為插空位擴充套件