https://medium.com/@monocosmo77/using-crank-nicolson-method-in-machine-learning-research-part3-3986ff8f51dd