Imported from https://github.com/iovdin/l4-pytorch