Tag: NUMBA
Estoy intentando inicializar un array “vacío” con cada elemento que contenga un array de t_list, un array de np.zeros de 8×8: t_list = np.zeros((8,8), dtype=np.float32) Ahora quisiera tener un np.array con múltiples t_list en cada índice: result = np.array((t_list, t_list, …., tlist)) Me gustaría poder controlar el número de veces . . . Read more
Tengo un kernel de CUDA y varias funciones de dispositivo en numba para un proyecto. Cuando trato de llamar a xoroshiro128p_uniform_float32 del módulo numba.cuda.random, siempre obtengo: import numba from numba import cuda from numba.cuda.random import create_xoroshiro128p_states from numba.cuda.random import xoroshiro128p_uniform_float64 @cuda.jit(‘void(float32[:,:])’, device=True) def device(rng_states): thread_id = cuda.grid(1) probability = xoroshiro128p_uniform_float64(rng_states, . . . Read more
Estoy enfrentando un problema para calcular una matriz de distancia grande. No obstante, esta es una matriz de distancia específica: es una matriz de puntos que están en una celda unitaria. Esta función obtiene coordenadas fraccionarias (entre 0 y 1 en todas las dimensiones) y me gustaría calcular la matriz . . . Read more
Tengo una función interpoladora f(x) que he creado usando interp1d. Ahora, a partir de la función f(x), debo construir una función f(y,z) donde x=(y-z)/2. Lo estoy haciendo usando un bucle anidado de la siguiente manera: f = lambda x: x**3 y = np.arange(0,100) z = np.arange(0,100) fnew = np.zeros((100,100)) for . . . Read more
Tengo dos matrices, una en 2-D (N x N) y la otra en 3-D (N x N x 3), respectivamente xi y R. Tengo que realizar el siguiente cálculo: h = 0 for i in range(N): h += xi[i].dot(R[i].dot(R[i].T).dot(xi[i])) Esto está incluido en una función decorada con numba (@njit). Estoy . . . Read more