MATLAB: Flow control with functions of matrices

if statementmatrix

theta is a matrix [0:0.01:0.5]*pi
I want to evaluate a function t using a flow control of values of theta:
if theta>pi*(1/2 - 0.01)
t=Inf
else
t = tan(theta)
end
How to achieve such a flow control when theta is a matrix? t is assumed to be a matrix with the same dimensions as theta.

Best Answer

t = inf(size(theta));
mask theta <= pi*(1/2 - 0.01);
t(mask) = tan(theta(mask));