Pytorch: RuntimeError: torch.cat(): input types can't be cast to the desired output type Byte

Input types can't be cast to the desired output type Long helped me resolve my issue.