def METHOD_NAME(model: torch.nn.Module, input_shapes: Union[Tuple, List[Tuple]]) \
        -> ConnectedGraph:
    """
    Create connected graph, using random inputs generated from given input shapes.
    :param model: torch model to create a connected graph from
    :param input_shapes: input shapes to the torch model
    :return: ConnectedGraph representation of the model
    """
    device = get_device(model)
    random_inputs = create_rand_tensors_given_shapes(input_shapes, device)
    return ConnectedGraph(model, random_inputs)