Identify a Matrix with PyTorch

Hi,

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 : https://colab.research.google.com/drive/1hngoL3R9G7jx3Yv-mTYMGRPFjbHiaaCC?usp=sharing

Thanks,

Thomas