Examples with DropoutLayer org.deeplearning4j.nn.conf.layers.DropoutLayer used on opensource projects

Example 1 with DropoutLayer

use of org.deeplearning4j.nn.conf.layers.DropoutLayer in project deeplearning4j by deeplearning4j.

the class DropoutLayerTest method testInputTypes.

@Test
public void testInputTypes() {
    DropoutLayer config = new DropoutLayer.Builder(0.5).build();
    InputType in1 = InputType.feedForward(20);
    InputType in2 = InputType.convolutional(28, 28, 1);
    assertEquals(in1, config.getOutputType(0, in1));
    assertEquals(in2, config.getOutputType(0, in2));
    assertNull(config.getPreProcessorForInputType(in1));
    assertNull(config.getPreProcessorForInputType(in2));
}