标签： python numpy numba
我有一个可矢量化的函数，我正在使用具有非标准签名的NumBa编写：它将长度为n的矢量映射到维度(n + 1, n)的数组中。据我所知，在NumBa中无法直接执行此操作：函数签名中不能包含算术，因此简单签名(n) -> (n+1, n)是不可能的。我发现了一个不优雅但可行的解决办法：我将此函数包装在另一个函数中，其中包含一个长度为m的伪参数，然后使用签名(n), (m) -> (m, n)。
n
(n + 1, n)
(n) -> (n+1, n)
m
(n), (m) -> (m, n)
我的问题是：是否有一种不那么脆弱的方法呢？