Imported from https://github.com/matwilso/relation-networks