Identify a Matrix with PyTorch


I am a beginner with PyTorch. I want to identify the matrix A in the ODE: dX/dt=A*X. So, I wrote a PyTorch code to do that. I only use a linear cell in the neural network.

Unfortunately for me, it doesn’t work at all. When I run the code, the loss is decreasing, but when I check the weights of the matrix, they are totally different from the coefficients of the matrix A.

Can someone help me?

The link of the notebook :