We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Hi. I suggest changing the following piece of code in the reach_star_single_input function of the FullyConnectedLayer.m class
reach_star_single_input
FullyConnectedLayer.m
V(1, 1, :, in_image.numPred + 1) = zeros(obj.OutputSize, 1, 'like', in_image.V); for i=1:n+1 I = in_image.V(:,:,:,i); I = reshape(I,N,1); % flatten input if i==1 V(1, 1,:,i) = obj.Weights*I + obj.Bias; else V(1, 1,:,i) = obj.Weights*I; end end
to this:
V(1, 1, :, :) = obj.Weights*reshape(in_image.V, N, n + 1); V(1, 1, :, 1) = reshape(V(1, 1, :, 1), obj.OutputSize, 1) + obj.Bias;
in order to improve efficiency for a large number of generators, for both cpu and gpu modes.
The text was updated successfully, but these errors were encountered:
No branches or pull requests
Hi. I suggest changing the following piece of code in the
reach_star_single_input
function of theFullyConnectedLayer.m
classto this:
in order to improve efficiency for a large number of generators, for both cpu and gpu modes.
The text was updated successfully, but these errors were encountered: