Currently I am training a physics informed neural network with the wave propagation data. The shape of