May I ask what does cumulative_entity_ids mean? when I ran it on cpu, it shows: RuntimeError: zero-dimensional tensor (at position 1) cannot be concatenated at https://github.com/YoungXiyuan/DCA/blob/master/mulrel_ranker.py cumulative_entity_ids = torch.cat([cumulative_entity_ids, entity_ids[indx][action.data[0]]], dim=0)