On second thoughts, for further use later on, when the user gives custom output_weights, when should the weights be given, after calculating the output r2_score for each dimension, or during the calcuation of numerator/denominator.
By the present code, it should be when calculating numerator/denominator
And according to @mblondel 's suggestion, it should be after calculating the r2_score for each dimension.