首页 > 解决方案 > 在 Julia 中,您可以指定可调用函数参数的参数和返回值吗?

问题描述

在 Python 中,您可以为传递给另一个函数(引用)的可调用对象指定参数和返回值。例如:

def foo(
    bar: Callable[[str],int], # a function that takes a string as an argument and returns an int 
    baz: List[str] # a list of strings
    ) -> List[int]: # returns a list of ints
    return [bar(s) for s in baz]

foo(len,["hello","world!"]) # apply the len function to the list of strings
# [5, 6]

我怎样才能Callable[[str],int]在 Julia 中写出等价的东西?

标签: typesjuliatype-hintingcallable

解决方案


在 Julia 中没有这样做的机制。Julia 和 Python 都不会静态地强制执行函数的类型签名。在 Julia 中,首选是让您的函数参数尽可能通用,因此大多数时候您最好将除函数之外的参数保留为通用,如下所示:

foo(f, s) = map(f, s)

尽管有时人们会f这样注释论点:

foo(f::Union{Function, Type}, s) = map(f, s)

Type包含在联合中是因为并非所有可调用对象都是 的子类型Function,例如,请查看手册中的Function-like objects

可以对函数的输出类型进行注释,但这实际上只是尝试将返回值转换为指定的输出类型,因此如果您的函数没有返回您所指定的类型,您可能会遇到运行时转换错误注释它。

输出类型的注释通常是多余的。考虑这段代码:

bar(x::Int)::Int = 2x

如果我们查看降低的代码,我们可以看到已添加到代码中的额外类型转换:

julia> @code_lowered bar(42)
CodeInfo(
1 ─ %1 = Main.Int
│   %2 = 2 * x
│   %3 = Base.convert(%1, %2)
│   %4 = Core.typeassert(%3, %1)
└──      return %4
)

但是,在这种情况下,编译器足够聪明,可以计算出当输入是整数时,输出将是整数并且不需要转换:

julia> @code_llvm bar(42)

;  @ REPL[1]:1 within `bar'
define i64 @julia_bar_788(i64) {
top:
; ┌ @ int.jl:87 within `*'
   %1 = shl i64 %0, 1
; └
  ret i64 %1
}

事实上,我们看到bar已经简化为向左移位操作。

julia> bar(42)
84

julia> 42 << 1
84

推荐阅读