00001 #ifndef DROPOUT_LAYER_H 00002 #define DROPOUT_LAYER_H 00003 00004 #include "layer.h" 00005 #include "network.h" 00006 00007 typedef layer dropout_layer; 00008 00009 dropout_layer make_dropout_layer(int batch, int inputs, float probability); 00010 00011 void forward_dropout_layer(dropout_layer l, network_state state); 00012 void backward_dropout_layer(dropout_layer l, network_state state); 00013 void resize_dropout_layer(dropout_layer *l, int inputs); 00014 00015 #ifdef GPU 00016 void forward_dropout_layer_gpu(dropout_layer l, network_state state); 00017 void backward_dropout_layer_gpu(dropout_layer l, network_state state); 00018 00019 #endif 00020 #endif