一个简单的例子:
import numpy as np
from numba import vectorize, float64
@vectorize(nopython=True)
def f(x,...@vectorize([float64(float64, float64)], nopython=True)
def f(x, y):
return x + y
还可以指定多个函数签名,需注意越通用的类型越要排在后面...import numpy as np
from numba import vectorize, int32, int64, float32, float64
@vectorize([int32(int32..., int32),
int64(int64, int64),
float32(float32, float32),
float64...(float64, float64)])
def f(x, y):
return x + y
此时如果传入其它类型,程序就会报错。