Imported from https://github.com/ars-ashuha/quantile-regression-dqn-pytorch