Imported from https://github.com/Driesssens/ibp-pytorch