This PR fixes the documentation of GeneralizedProximalGradient to match the code implementation.
Moreover, small changes are introduced to the VStack code to enforce consistency between the expected size of x (as given by nn or restr) and the actual size of the array.
This PR fixes the documentation of GeneralizedProximalGradient to match the code implementation.
Moreover, small changes are introduced to the VStack code to enforce consistency between the expected size of x (as given by
nn
orrestr
) and the actual size of the array.