Skip to content

Conversation

@marceloribeiro19
Copy link

This PR adds a fix for the edge case where num_o_neurons == 1.

The original code returns invalid results for a single output neuron, as the optimized SIMD loop expects at least 2 neurons.

-This patch adds a simple scalar fallback.
-Tested manually against expected values.
-Backward-compatible.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant