Может быть, вы можете извлечь некоторые идеи из следующих комментариев:
1) Рассчитать лог (W [i, j]) с упрощением формулы, показатели исчезают, обработка должна быть быстрее .
2) Возьмите его показатель степени: Exp (Log (W {i, j])) == W [i, j]
3) Используйте переменные для значений, которые являются постоянными внутри итераций, таких как sigma = 2 * sigma ** 2, которые вы можете вычислить при запуске вне итераций.
Важно, перед любым изменением запомнить результат, чтобы ваша новая разработка могла быть протестирована в финальной версии. Матричный результат, который вы уже знаете, я полагаю, правильный.
Удачи.