在slim.stack中交織slim.dropout和slim.fully_connected？

Question

在tf.slim中，我想創建一個完全連接的圖層，並將其丟棄。

以文檔爲例： slim.stack(x, slim.fully_connected, [32, 64, 128], scope='fc')，我想添加dropout。

是否可以使用slim.stack或我必須回到詳細的方法？

(pseudo-code) for every layer: 
    slim.dropout(slim.fully_connected(...) 

Answer 1

請看代碼：https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/layers/python/layers/layers.py#L1976。您需要具有以下特徵的局部功能：

outputs = layer(outputs, *layer_args, **kwargs) 

Answer 2

基於@ guinny的評論，我可以創造一個局部功能：

layer = lambda inputs, layer_args, **kwargs:  
    slim.dropout(slim.fully_connected(inputs, layer_args, **kwargs)) 
with slim.arg_scope([slim.dropout], keep_prob=dropout_keep_prob): 
    layers = slim.stack(inputs, 
         layer, 
         layer_sizes, 
         activation_fn=nonlinearity, 
         scope=scope)