Сравнение Туллио с numpy.einsum

#python #julia #numpy-einsum

Вопрос:

В настоящее время я импортирую скрипт Julia, который использует Tullio из-за его скорости. Функция заключается в

 using Tullio, LoopVectorization

function testfunction_tullio(my_arr, other_arr, sec_arr, third_arr)
    new_array = Array{ComplexF64}(undef, length(my_arr))
    @tullio new_array[i] = 1 / (my_arr[i] * 1im   sec_arr[j]   other_arr[i,1]) * third_arr[j]
    return new_array
end
 

Есть ли хороший способ написать это numpy.einsum() ? Все это 1D массивы ( или 1D компонент 2D массива), но я не знаю, как включить инверсии или что-то в этом роде.

Комментарии:

1. Я не знаю, как насчет numpy, но вам не нужно предварительно выделять new_array . Просто напиши @tullio new_array[i] := ...