(col, row, rank)

()

(u0 : str, nu, rho, nx = 256, nt = 100)

(u0: str, rho, nx=256, nt=100)

(X_all, N)

(u0: str, nu, beta, source=0, xgrid=256, nt=100)

(u0: str)

(model)

(x, t, beta, nu, rho, net)

(self, hidden_dim)

(u0 : str, beta, nu, rho, nx = 256, nt = 100)

(x, t, beta, nu, rho, net, hidden_dim)

()

()

()

()

(seed)

(self, hidden_dim, start_w, start_b, end_w, end_b, col_0, col_1, col_2, row_0, row_1, row_2,

(self, x, t, beta, nu, rho)

(self, x, t)

(coeff, meta_layer_1_w, meta_layer_1_b, meta_layer_2_w, meta_layer_2_b, meta_layer_3_w, meta_layer_3_b,

(activation)