flow control with functions of matrices

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.

Answer 1

[Walter Roberson]

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