Решить PDE с помощью Tensorflow

#tensorflow

#тензорный поток

Вопрос:

Я хотел бы знать, возможно ли решить 1D уравнение теплопроводности с помощью Tensorflow. Уравнение имеет вид:

введите описание изображения здесь

Возможно ли преобразовать следующий код python во входные данные Tensorflow? Большое вам спасибо.

 # Initial and boundary conditions 
T = np.ones(shape=(mesh_num 1,1))*T_amb
T[0] = 0.
T[mesh_num] = 0.

# Solve discrete equation 
for k in range(Total_itr_num):
    epsT = np.sin(3.0*np.pi/200.0*T)
    source = epsT *(T_amb**4-T**4)   h*(T_amb-T)

    for ii in range(mesh_num-1):
        i = ii   1
        T[i] = 0.5*(T[i 1] T[i-1] source[i]*dx*dx)