%% Include Layer in Network
% You can use a custom layer in the same way as any other layer in Deep Learning Toolbox.
% Define a custom PReLU layer. To create this layer, save the file preluLayer.m in the current folder.
% Create a layer array that includes the custom layer preluLayer.

layers = [
    imageInputLayer([28 28 1])
    convolution2dLayer(5,20)
    batchNormalizationLayer
    preluLayer(20,'prelu')
    fullyConnectedLayer(10)
    softmaxLayer
    classificationLayer];

% Copyright 2018 The MathWorks, Inc.