convert_dict_values_to_tensors

lightning_pose.utils.pca.convert_dict_values_to_tensors(param_dict: dict[str, ndarray | float], device: str | device) → dict[str, Tensor][source]

Convert all values in a parameter dictionary to float tensors on the given device.

Parameters:

  • param_dict – mapping from parameter name to scalar or array value.

  • device – target device for the output tensors.

Returns:

Dictionary with the same keys and values converted to torch.float tensors.