Function get_tensor

python/paddle/framework/io_utils.py:185–188

(var)

@static_only
def _legacy_static_save(param_dict, model_path, protocol=2):
    def get_tensor(var):
        if isinstance(var, (paddle.Tensor, core.DenseTensor)):
            return np.array(var)
        return var

    param_dict = {name: get_tensor(param_dict[name]) for name in param_dict}