How to solve nan loss in attention mechanism for multi-step forecasting?

Question

I have tried to develop the multi-step forecasting algorithm at nuclear power plants using the attention mechanism. This network outputs the multi parameters as well as mult