es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Tag: NUMBA

Llamando a una función jitted de otro archivo Python

Tengo un proyecto en Python con la siguiente estructura de documentos: /scripts main.py /classes myclass.py … La clase definida en myclass.py tiene un método que pretendo acelerar. He logrado esto utilizando “jitting” en algunas partes, convirtiéndolas en funciones de modo que myclass.py quede de la siguiente manera: def myclass: def . . . Read more

Optimización de Numba

Tengo una función de Python lenta y no puedo hacerla más rápida con numba. ¿Qué debo hacer? char_vectors_numba = Dict.empty( key_type=types.unicode_type, value_type=types.float64[:], ) X = np.zeros((1, 128, len(chars)), dtype=np.bool) @njit() def get_matrices(sentences, x=X[:]): max_sentence_len = 128 for i, sentence in enumerate(sentences): char_seq: str = (START_CHAR + sentence + END_CHAR).ljust(max_sentence_len+1, PADDING_CHAR) . . . Read more

Numba compila las funciones AOT con argumentos funcionales.

Estoy tratando de compilar AOT en Numba una función que tiene un argumento funcional, pero no puedo encontrar una forma de especificar correctamente su firma. Usando un ejemplo muy básico, con el decorador estándar de numba @njit, escribiría: import numba as nb @nb.njit(nb.f8(nb.f8, nb.f8)) def fcn_sum(a, b): return a + . . . Read more