Для оптимизации np.unravel_index можно использовать функцию np.argsort, которая возвращает индексы отсортированных элементов массива. Затем можно извлечь первый индекс и преобразовать его в двумерные координаты с помощью np.unravel_index.

Пример оптимизации:

sorted_indices = np.argsort(np.abs(out_array - ty) + np.abs(out_array2 - tx))
index = np.unravel_index(sorted_indices[0], out_array.shape)

Этот код избавляет от лишнего вызова np.abs и argmin, что может ускорить выполнение операции. Надеюсь, это поможет вам оптимизировать поиск индексов в массиве.