@@ -520,12 +520,12 @@ function set_name(obj, name)
520520
521521 % compute output sets
522522 if ~isa(input .V , ' gpuArray' )
523- % c = vl_nnconv(input.V(:,:,:,1), obj.Weights, obj.Bias, 'Stride', obj.Stride, 'Pad', obj.PaddingSize, 'Dilate', obj.DilationFactor);
524- % V = vl_nnconv(input.V(:,:,:,2:input.numPred + 1), obj.Weights, [], 'Stride', obj.Stride, 'Pad', obj.PaddingSize, 'Dilate', obj.DilationFactor);
525- c = dlconv(dlarray(input .V(: ,: ,: ,1 ), " SSC" ), obj .Weights , obj .Bias , ' Stride' , obj .Stride , ' Padding' , obj .PaddingSize , ' DilationFactor' , obj .DilationFactor );
526- V = dlconv(dlarray(input .V(: ,: ,: ,2 : input .numPred + 1 ), " SSCB" ), obj .Weights , 0 , ' Stride' , obj .Stride , ' Padding' , obj .PaddingSize , ' DilationFactor' , obj .DilationFactor );
527- c = extractdata(c );
528- V = extractdata(V );
523+ c = vl_nnconv(input .V(: ,: ,: ,1 ), obj .Weights , obj .Bias , ' Stride' , obj .Stride , ' Pad' , obj .PaddingSize , ' Dilate' , obj .DilationFactor );
524+ V = vl_nnconv(input .V(: ,: ,: ,2 : input .numPred + 1 ), obj .Weights , [], ' Stride' , obj .Stride , ' Pad' , obj .PaddingSize , ' Dilate' , obj .DilationFactor );
525+ % c = dlconv(dlarray(input.V(:,:,:,1), "SSC"), obj.Weights, obj.Bias, 'Stride', obj.Stride, 'Padding', obj.PaddingSize, 'DilationFactor', obj.DilationFactor);
526+ % V = dlconv(dlarray(input.V(:,:,:,2:input.numPred + 1), "SSCB"), obj.Weights, 0, 'Stride', obj.Stride, 'Padding', obj.PaddingSize, 'DilationFactor', obj.DilationFactor);
527+ % c = extractdata(c);
528+ % V = extractdata(V);
529529 else
530530 c = dlconv(dlarray(input .V(: ,: ,: ,1 ), " SSC" ), obj .Weights , obj .Bias , ' Stride' , obj .Stride , ' Padding' , obj .PaddingSize , ' DilationFactor' , obj .DilationFactor );
531531 V = dlconv(dlarray(input .V(: ,: ,: ,2 : input .numPred + 1 ), " SSCB" ), obj .Weights , 0 , ' Stride' , obj .Stride , ' Padding' , obj .PaddingSize , ' DilationFactor' , obj .DilationFactor );
0 commit comments