def _np2paddle()

in bindings/python/py_src/safetensors/paddle.py [0:0]

• 4 lines of code
• 2 McCabe index (conditional complexity)

def _np2paddle(numpy_dict: Dict[str, np.ndarray], device: str = "cpu") -> Dict[str, paddle.Tensor]:
    for k, v in numpy_dict.items():
        numpy_dict[k] = paddle.to_tensor(v, place=device)
    return numpy_dict