TypeError: torch.FloatTensor constructor received an invalid combination of arguments - got (int, int, numpy.int64, numpy.int64), but expected one of:
* no arguments
* (int ...)
didn't match because some of the arguments have invalid types: (int, int, numpy.int64, numpy.int64)
* (torch.FloatTensor viewed_tensor)
* (torch.Size size)
* (torch.FloatStorage data)
* (Sequence data)
Which version of pytorch is to be used?