TypeError: full() received an invalid combination of arguments - got (tuple, int, device=NoneType),