customing a layer and error

Question

JaeHyeon Lee 2021-1-28

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/729393-customing-a-layer-and-error

编辑： JaeHyeon Lee 2021-2-10

After customizing the layer, I used the checkLayer function to check the validity of the layer. If the input is (H, W, C) and debugging, the input is expanded to (H, W, C, N). Can you tell me why this is so?

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Shubham Khatri 2021-2-3

It should error out on entering the third parameter as you have not specified its value in the fourth variable. Can you share the reporduction steps?

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Shubham Khatri 2021-2-3

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/729393-customing-a-layer-and-error#answer_613637

在 MATLAB Online 中打开

Hello,

To my understanding the checklayer funtion accepts input in 2 ways.

2 inputs-

checkLayer(layer,validInputSize)

Checks the validity of a custom layer using generated data of the sizes in validInputSize. Variable validInputSize defines the number of inputs to the layer.

4 inputs-

checkLayer(layer,validInputSize,Name,Value)

The third input and fourth input are linked with each other. The third input defines the additional parameter for check and the fourth input defines the value of the third parameter to compare to.

Please refer to the documentation link for checklayer here

Hope it helps.

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

JaeHyeon Lee 2021-2-10

编辑：JaeHyeon Lee 2021-2-10

在 MATLAB Online 中打开

I made the source code of the layer like this, and when debugging, I could see that the shape of X1, X2, X3 is changed to 4D-dlarray. Can I know specifically why?

The validity of the layer was checkedLayer(layer, {[28 28 128], [28 28 128], [28 28 128]},'ObservationDimension', 4).

classdef nonLocalBlock < nnet.layer.Layer
    properties (Learnable)
        % Layer learnable parameters
        % Scaling coefficients
    end
    
    methods
        function layer = nonLocalBlock(numInputs,name) 
            layer.NumInputs = numInputs;
            layer.Name = name;
            layer.Description = "Non-local Block" ;
        end
        
        function Z = predict(~, X1, X2, X3)    
            X1_size = size(X1);
            X2_size = size(X2);
            X3_size = size(X3);
            [H, W, C, D] = size(X1); 
            
            X1 = reshape(X1 ,H*W, C, 1, []);
            X2 = reshape(X2, C, H*W, 1, []);
            X3 = reshape(X3 ,H*W, C, 1, []);
            
            if D > 1
               f = pagemtimes(X1, X2);
            else
               f = mtimes(X1, X2);
            end
          
            f_div_C = softmax(f,'DataFormat','SSCB');
        
            if D > 1
               y = pagemtimes(f_div_C, X3);
            else
               y = mtimes(f_div_C, X3);
            end
            
            y = reshape(y, H, W, C, []);
            Z = y;
        end
    end
end