Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def from_numpy(cls, *nparrs) -> 'BlobTensor':
blobs = []
for arr in nparrs:
blobs.append(arr.data)
dt = DType.__members__[str(nparrs[0].dtype)]
return cls(dt, nparrs[0].shape, *blobs)
def _to_numpy_type(t):
if isinstance(t, DType):
t = t.value
mm = {
'FLOAT': 'float32',
'DOUBLE': 'float64'
}
if t in mm:
return mm[t]
return t.lower()