How to convert it to a torch tensor?