Imported from https://github.com/gusye1234/Pytorch-HSIC-bottleneck