I should either be an MxN, MxNxP, MxNx1xP or an MxNx3xP sized array.

Hi, i am trying to visualizate second convolutional layer's filter weights in my neural network but i get this error. For first conv layer it worked. Here is the code for visualization.

% Extract the first convolutional layer weights
w = cifar10Net.Layers(2).Weights;

% rescale the weights to the range [0, 1] for better visualization
w = rescale(w);

figure
montage(w)

2 is replaced with 6 in my case. This is the error

Error using montage>parse_inputs (line 255)
I should either be an MxN, MxNxP, MxNx1xP or an MxNx3xP sized array.

Error in montage (line 179)
    parse_inputs(varargin{:});

Please help