#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] := ...