如果存在多个相同类型的lambda，如何返回特定的lambda？(KOTLIN)

val lambda1: (Int) -> Int = { x -> x * 2 }
val lambda2: (Int) -> Int = { x -> x * x }

fun returnSpecificLambda(label: String): (Int) -> Int {
    return when (label) {
        "lambda1" -> lambda1
        "lambda2" -> lambda2
        else -> throw IllegalArgumentException("Invalid label")
    }
}

val specificLambda = returnSpecificLambda("lambda1")
val result = specificLambda(5) // 使用特定的lambda进行计算

val lambda1: (Int) -> Int = { x -> x * 2 }
val lambda2: (Int) -> Int = { x -> x * x }

fun returnSpecificLambda(lambda: (Int) -> Int): (Int) -> Int {
    return when (lambda) {
        lambda1 -> lambda1
        lambda2 -> lambda2
        else -> throw IllegalArgumentException("Invalid lambda")
    }
}

val specificLambda = returnSpecificLambda(lambda1)
val result = specificLambda(5) // 使用特定的lambda进行计算