#python #numba
Вопрос:
Я сталкиваюсь с проблемой при использовании пакета numba python. Мой код выглядит следующим образом:
@jit() def key_gen(index,x,HKu): H = HKu[index-1][0][0:-1] H = H.astype('float64') K = HKu[index-1][0][-1] K = K.astype('float64') d = H@x-K if d gt;= 0.0: key = index else: key = -index return key`
У меня есть несколько предупреждений, которые я не понимаю:
Compilation is falling back to object mode WITH looplifting enabled because Function "key_gen" failed type inference due to: Invalid use of BoundFunction(array.astype for array(uint8, 1d, C)) with parameters (Literal[str] (float64))
type(index) = int,type(x) = numpy array,type(HKu) = list of numpy array
Пожалуйста, помогите мне правильно использовать NumPy и Numba. Спасибо
Комментарии:
1. Ошибка заключается в том, что он не знает, что делать, когда вы передаете строку в
astype
метод. Возможно , попробуйте пройтиnp.float64
, т. е.H.astype(np.float64)
.