1. 程式人生 > >tensor flow 學習 tf.contrib.layers.flatten()和tf.contrib.layers.fully_connection()

tensor flow 學習 tf.contrib.layers.flatten()和tf.contrib.layers.fully_connection()

tf.contrib.layers.flatten(P)這個函式就是把P保留第一個維度,把第一個維度包含的每一子張量展開成一個行向量,返回張量是一個二維的, shape=(batch_size,….),一般用於卷積神經網路全連結層前的預處理
t f.contrib.layers.fully_connection(F,num_output,activation_fn)這個函式就是全連結成層,F是輸入,num_output是下一層單元的個數,activation_fn是啟用函式