Physics-informed attention-based neural network for solving non-linear partial differential equations