Imported from https://github.com/makora9143/pytorch-convcnp