switch to torch.concat and torch.unique instead of list.append and set()

switch to torch.concat and torch.unique instead of list.append and set() #68