How to vectorize function on vectors stored in 3D tensor

Hi everyone.

I have a function that sums the 3rd dimension as presented here.

This is the code:

function sumeol = sum_over_line(tau, L, zstruct, rstruct, P)
before = zeros(P.Ntimes_An, rstruct.rlength);
for i=1:P.Ntimes_An
    for j=1:rstruct.rlength
        for k=1:zstruct.zlength-2
            before(i, j) = before(i, j).*(1-1./(exp(tau(i, j, k)))) + L(i, j, k);
        end
    end
end
sumeol = before;
end

I would like to vectorize it - meaning getting rid of the loop over i and j and just having one loop so that the function will get the tensors tau and L sized aXbXc and return a matrix aXb.

How should I go about that? Thank you in advance

[Dyuman Joshi]

There was a misunderstanding on my side, Sorry. I have posted a new answer, please check it.

Answer 1

[Dyuman Joshi]

Try this -

function sumeol = sum_over_line(tau, L, zstruct, rstruct, P)
before = zeros(P.Ntimes_An, rstruct.rlength);
i=1:P.Ntimes_An;
j=1:rstruct.rlength;
    for k=1:zstruct.zlength-2
        before(i, j) = before(i, j).*(1-1./(exp(tau(i, j, k)))) + L(i, j, k);
    end
sumeol = before;
end

[Noya Linder]

Thank you so much!!

[Dyuman Joshi]

You are welcome!