new pytorch datatype support

hbgtjxzbbx

The new pytorch(0.4) change its data type, so some fixes need to be done.