According to the formula of REINFORCE, we could define the loss function as log-probability of actions multiplied by the rewards. I am curious if it is to replace action.reinforce(r) as loss = outputs of final layer * rewards ?