dropout functionality moved to layer class