Hi,
You need to keep in mind that this module expects a batch of inputs.
And so in the forward pass, the bias is actually automatically broadcasted over all the elements in the batch. In the backward pass, we need to do the backward of this broadcast on the 0th dimension which is a sum over this 0th dimension.