def tensors_to_device()

in summarize_from_feedback/utils/torch_utils.py [0:0]

• 9 lines of code
• 5 McCabe index (conditional complexity)

def tensors_to_device(data, device):
    if data is None:
        return None
    elif isinstance(data, torch.Tensor):
        return data.to(device)
    elif isinstance(data, dict):
        return {k: tensors_to_device(v, device) for k, v in data.items()}
    else:
        raise ValueError(f"Unsupported type: {type(data)}")