Hi,
What do you want to accomplish with this?
Also, could you please specify the variables a_i and x_i you’re using?
This might be worth checking out to make sure that you actually need the second set of weights About BCEWithLogitsLoss’s pos_weights.