-
Notifications
You must be signed in to change notification settings - Fork 3
/
utils.py
33 lines (28 loc) · 819 Bytes
/
utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import numpy as np
# the overflow and underflow limit in tf.float32.
OVERFLOW_LIMIT = 1e38
UNDERFLOW_LIMIT = 1e-37
OVERFLOW_D = 38
UNDERFLOW_D = -37
# onverts data types in numpy to python primitive data types.
def resolve_type(y):
if isinstance(y, np.int32) or isinstance(y, np.int64):
return int(y)
elif isinstance(y, np.float32) or isinstance(y, np.float64):
return float(y)
elif isinstance(y, np.bool):
return bool(y)
else:
return y
# parses the tensor shape from protocol buffer file into a python list
def shape_from_proto(shape):
s = str(shape)
x = 0
u = []
for i in range(len(s)):
if s[i] >= '0' and s[i] <= '9':
x = x * 10 + ord(s[i]) - 48
elif x != 0:
u.append(x)
x = 0
return u