Comment vectoriser l'entrée de fonction de matlab quand NaN pourrait être l'entrée?

Question

j'ai écrit une fonction que j'espère avoir l'entrée et la sortie vectorisé:

function output = myfunction(input1,input2) 
if input1 == 0 
    output = equation1 ; 
else 
    output = equation2 ; 
end 

Je l'ai testé avec input1 = 0 et input1 = 0.5 et myfunction fonctionne bien. input1 et input2 sont en réalité des matrices 3D. J'essaie d'avoir un calcul plus efficace au lieu d'exécuter une boucle pour un calcul élément par élément. Si input1 sont tous les éléments non-zéro, myfunction fonctionne correctement. Il semble qu'une fois que j'ai entré une matrice avec des éléments zéro et non nul, il ne peut pas passer par le if-statement correctement. Ma fonction me donnera NaN pour l'élément qui a input1==0.

Comment puis-je le modifier afin que myfunction peut prendre un nombre ou une matrice dans un if-statement?

Merci!

[EDIT:]

La fonction réelle ressemble à ceci:

function output = myfunction(input1 , frequency , input2 , input3) 

zero_ind = input1 == 0 ; 
output = zeros(size(input1)) ; 
output(zero_ind) = input3(zero_ind)/frequency ./ input2(zero_ind) ; ; 
temp1(~zero_ind) = input1(~zero_ind)/frequency ; 
temp2(~zero_ind) = input2(~zero_ind) * frequency ; 
output(~zero_ind) = input3(~zero_ind) .* (1 + temp1(~zero_ind)).^temp2(~zero_ind) .* temp1(~zero_ind) ./ ((1 + temp1(~zero_ind)).^temp2(~zero_ind) - 1) ; 

Answer 1

Voici une suggestion générale de vectorisation. Cela devrait être changé correctement à votre type de calculs:

function output = myfunction(input1,input2) 
zero_ind = input1==0; % this replaces the if statement 
output = zeros(size(input1)); % initialize output to wanted size and shape 
output(zero_ind) = input2(zero_ind).^2; % equation1 
output(~zero_ind) = input1(~zero_ind).*input2(~zero_ind); % equation2 
end 

L'idée principale est d'utiliser logical indexing au lieu de la déclaration if. Je suppose ici (entre autres, je suppose) que toutes vos entrées et sorties sont de la même taille. Cela pourrait être assoupli, mais cela dépend du calcul que vous faites.

EDIT:

En réponse à votre édition avec la fonction réelle, voici ma suggestion (j'ai changé les noms d'entrée juste pour le rendre un peu plus compact):

function output = myfunction(in1,frequency,in2,in3) 
temp1 = in1/frequency ; 
temp2 = in2*frequency ; 
output = in3.*(1+temp1).^temp2.*temp1./((1+temp1).^temp2-1); 
zero_ind = in1 == 0; 
output(zero_ind) = in3(zero_ind)./frequency./in2(zero_ind); 
end 

Ce que je l'ai fait est de calculer tous output comme il n'y a pas de zéros dans in1, puis remplacez uniquement les endroits dans output qui correspond à un zéro dans in1 à une autre valeur. De cette façon, vous n'utilisez zero_ind que lorsque cela est nécessaire et minimisez l'opération d'indexation.