How to have elementwise multiplication expand in two dimensions?

2 vues (au cours des 30 derniers jours)
AlexRD
AlexRD le 24 Juin 2022
Commenté : AlexRD le 25 Juin 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!

Réponse acceptée

Matt J
Matt J le 24 Juin 2022
Modifié(e) : Matt J le 24 Juin 2022
C=A.*reshape(B,[16,16,3,1,7]);
C=reshape(C,[16,16,3,700]);

Plus de réponses (0)

Catégories

En savoir plus sur Execution Speed dans Help Center et File Exchange

Community Treasure Hunt

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

Start Hunting!

Translated by