You need to call .reinforce() on outputs of stochastic function (.bernoulli(), .normal() and .uniform() at the moemnt), if you want to have autograd estimate the gradient of the expectation of the reward. No need to use it if you don’t do any sampling.