Defining complex initial conditions

[Mide2001]

I want to evaluate PINNs and compare them to numerical methods for engineering applications and one of these applications is the 2D heat equation. So I'm trying to set up a complex initial condition(IC) over a square domain. The IC that I want to set up is an annulus. I'm able to do this with basic code but I can't figure out how to do this with DeepXDE. I want to be able to define the IC, in this case, an annulus, and the magnitude at each point in this case the temperature. Is there a way to do this?

Here's the code for the basic solution:

for i in range(nx):
for j in range(ny):
if ( ( (idx-1)**2+(jdy-1)**2 <= 0.3)& ((idx-1)**2+(jdy-1)**2>=.1) ):
u[i,j] = 2.1
else:
u[i,j] = 1

And here's what i have in DeepXDE so far:

def func(x):
return -((((x[:, 0:1]-2.5)**2)+((x[:, 1:2]-2.5)**2)) -0.3)

[lululxvi]

See FAQ "Q: More details and examples about initial conditions." and "Q: More details and examples about boundary conditions."

[Mide2001]

I've checked the FAQ's and unfortunately couldn't find any answers. Is there a way to assign a value to the initial conditions?

[lululxvi]

The func at https://deepxde.readthedocs.io/en/latest/modules/deepxde.icbc.html#deepxde.icbc.initial_conditions.IC can be any function, no matter how complex it is.