CarlosHernandezP:

So it does make sense that the size equals my output size but I’m still left unsure as to whether my method updates the bias correctly or not.


If you are not changing the number of output features, you could just reuse the bias.
Your approach should work, besides the usage of .data which should be avoided.