How to vectorize function on vectors stored in 3D tensor

Question

Noya Linder 2023-10-1

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/2027804-how-to-vectorize-function-on-vectors-stored-in-3d-tensor

评论： Dyuman Joshi 2023-10-3

采纳的回答： Dyuman Joshi

在 MATLAB Online 中打开

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

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Dyuman Joshi 2023-10-2

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

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Dyuman Joshi 2023-10-2

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/2027804-how-to-vectorize-function-on-vectors-stored-in-3d-tensor#answer_1323329

在 MATLAB Online 中打开

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