python中,定义一个pde函数,传入的数据维度可以是不一样的

在写FT_PINNs时,遇到问题,因为对x进行FFT变换,所以pde中的x成为了参数,我做了一个测试,

tBox = [0, 2]
t_eqn = np.random.uniform(tBox[0], tBox[1], size=(1000,))  # t = tBox[0] ~ tBox[1]

x =  2.4609375
k = 4.948008429403925

def u0(self, t, x, k):
    val = 6 *  np.exp(-t) * (np.exp(x) - np.exp(-x)) / 2
    return val
c = u0(t_eqn, x ,k)

# 对于FFT变换不影响

d = np.fft.fft(u0(t_eqn, x, k))

 c.shap:(1000,) d.shape(1000, )