How to vectorize function on vectors stored in 3D tensor

3 次查看（过去 30 天）
Noya Linder
Noya Linder 2023-10-1
评论： Dyuman Joshi 2023-10-3
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
Dyuman Joshi 2023-10-2
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

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Loops and Conditional Statements 的更多信息

标签

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by