Imported from https://github.com/pfnet-research/einconv