create fully connected layer scratch in matlab

main.m
classdef FullyConnectedLayer < nnet.layer.Layer
    
    properties
        InputSize
        OutputSize
        Weights
        Bias
    end
    
    methods
        function layer = FullyConnectedLayer(outputSize, name)
            layer.Name = name;
            layer.OutputSize = outputSize;
        end
        
        function layer = initialize(layer, inputSize)
            layer.InputSize = inputSize;
            layer.Weights = randn(layer.OutputSize, layer.InputSize);
            layer.Bias = randn(layer.OutputSize, 1);
        end
        
        function Z = predict(layer, X)
            Z = layer.Weights*X + layer.Bias;
        end
    end
end
665 chars
27 lines

main.m
layer = FullyConnectedLayer(10, 'fc');
net = nnet.SequentialLayer();
net.add(layer);
85 chars
4 lines