Возможно, это тривиально, но мне было интересно, как писать сигнатуры в декораторе jit
, когда есть несколько выходов.
Например:
import numba as nb
@nb.jit(['???(int32, int32, float(:,:), float(:,:))'], nopython=True)
def foo(nx, ny, a, b):
for i in range(nx):
for i in range(ny):
do stuff with a & b
return a, b
Как насчет выступлений? Лучше ли писать две разные функции?