Spaces:
Sleeping
Sleeping
import numpy as np | |
import torch | |
def v_wrap(np_array, dtype=np.float32): | |
if np_array.dtype != dtype: | |
np_array = np_array.astype(dtype) | |
return torch.from_numpy(np_array) | |