function that checks multiple numbers

%Greetings, Id like to write a function that will check multiple numbers in a %matrix insted of a single number. For example I have this function

function [output] = my_func(input)
output=[]
if mod(sqrt(input),1)==0
    output=sqrt(input)
else 
    output=floor(input/3)
end

%but Id like to make it check multiple numbers instead of one?

[madhan ravi]

doc varargin
doc varargout

Answer 1

[KSSV]

function [output] = my_func(input)
output = zeros(size(input)) ;
c = mod(sqrt(input,1)) ;
idx = c==0 ;
output(idx) = sqrt(input(idx)) ;
output(~idx) = floor(input(~idx)/3) ; 

You need to re think on this line:

mod(sqrt(input),1)==0

[madhan ravi]

idx = c==0 ; % is completely redundant, there is no need for "==" equality

[Reynaldo Ponce]

sounds good sir. So i'd like to check a whole matrix or two numbers or five numbers like this...

my_func(1,2,81)

[KSSV]

make input a matrix.