Imported from https://github.com/naoto0804/pytorch-inpainting-with-partial-conv