Update file util.py

util.py

@@ -1,6 +1,7 @@
 import json
 import math
 import random
+import torch