How to have elementwise multiplication expand in two dimensions?

2 visualizaciones (últimos 30 días)
AlexRD
AlexRD el 24 de Jun. de 2022
Comentada: AlexRD el 25 de Jun. de 2022
So I'm trying to optimize my code, and one aspect where it could really improve is to change one for loop to a SIMD instruction.
In short, I have two variables: A (16 x 16 x 3 x 100) and B(16 x 16 x 3 x 7)
What I want to do is to get an elementwise multiplication that expands to both the 100 and 7 dimensions, concatenating and returning a variable of size C (16 x 16 x 3 x 700).
If i do A .* B(:, :, :, 1) it automatically expands to the 100, but then i'd need a 7 iteration for loop concatenating along B.
How do i do this without a for loop?
Thanks!

Respuesta aceptada

Matt J
Matt J el 24 de Jun. de 2022
Editada: Matt J el 24 de Jun. de 2022
C=A.*reshape(B,[16,16,3,1,7]);
C=reshape(C,[16,16,3,700]);

Más respuestas (0)

Categorías

Más información sobre Execution Speed en Help Center y File Exchange.

Community Treasure Hunt

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

Start Hunting!

Translated by