Imported from https://github.com/AndreaCossu/Relation-Network-PyTorch