* Update torch_utils.py

fix issue#113 , inputs device should be same with model parameters' device

* Update torch_utils.py

* Update torch_utils.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>