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)