

In [1]:

    
# Необходмые команды импорта.
import sys
sys.path.append('../../physlearn/')
sys.path.append('../../source')
import numpy as np
from numpy import linalg as LA
import tensorflow as tf
from matplotlib import pylab as plt
import numpy.random as rand
from physlearn.NeuralNet.NeuralNet import NeuralNet
from physlearn.Optimizer.NelderMead.NelderMead import NelderMead
import d1_osc
import ann_constructor
import math_util
from lagaris2d import LagarisSolver2d

from lagaris_mc import LagarisSolver_mc

from visualiser import Visualiser
from mpl_toolkits.mplot3d.axes3d import Axes3D
%matplotlib inline

def grid_vect(a,b,N):
    x = np.linspace(a, b, N, endpoint=True)
    h = x[1] - x[0]
    return np.mgrid[a:b+h:h, a:b+h:h].reshape(2,-1).T

def get_meshgrid(a,b,N):
    x = np.linspace(a, b, N, endpoint=True)
    return np.meshgrid(x,x)

n_sig = 12
a = -5
b = 5
#m_train = 35
#m_trapz = 25
#train_x = np.linspace(a, b, m_train, endpoint = True).reshape(1, m_train) 
#trapz_x = np.linspace(a, b, m_trapz, endpoint = True)
#obs_x = np.linspace(a, b, 200, endpoint = True).reshape(1, 200)

N = 60
xy = grid_vect(a,b,N)
xg,yg = get_meshgrid(a,b,N)

train_xy = np.transpose(xy)
trapz_xy = train_xy









    



D:\Anaconda\lib\site-packages\h5py\__init__.py:36: FutureWarning: Conversion of the second argument of issubdtype from `float` to `np.floating` is deprecated. In future, it will be treated as `np.float64 == np.dtype(float).type`.
  from ._conv import register_converters as _register_converters



In [2]:

    
lagar = LagarisSolver_mc(2)
lagar.define_psi(n_sig)
net_x = lagar.get_net_x()
dim = lagar.get_dim()
sess = lagar.get_sess()

V = tf.square(net_x)
lagar.define_H_psi(V)

first_deriative = tf.gradients(lagar.psi, lagar.net.x)[0] grad2 = tf.gradients(first_deriative, lagar.net.x)[0] U_psi = tf.multiply(tf.reduce_sum(V, axis=0), lagar.psi) T_psi = (-tf.reduce_sum(grad2, axis=0)) H_psi = T_psi + U_psi bbb = lagar.net.calc(H_psi, {lagar.net.x:np.transpose(xy), lagar.beta: 1}) print(bbb) bbb.shape



In [3]:

    
Z = lagar.calc_psi(np.transpose(xy), 1)
print(Z)
print(Z.shape)
fig = plt.figure(figsize=(14,6))
ax = fig.add_subplot(1, 2, 1, projection='3d')
xg,yg = np.meshgrid(xy[:,0], xy[:,1])
ax.plot_wireframe(xy[:,0], xy[:,1], Z)









    



[[-3.34053112e-22 -1.74472760e-21 -8.61346886e-21 ...  8.95079680e-23
   1.63437507e-23  2.80172662e-24]]
(1, 3721)






    Out[3]:





<mpl_toolkits.mplot3d.art3d.Line3DCollection at 0x1e9373922b0>



In [4]:

    
Hpsi = lagar.calc_H_psi(np.transpose(xy), 1)
#Hpsi = Hpsi.reshape(1,Hpsi.size)
print(Hpsi)
print(Hpsi.shape)
fig = plt.figure(figsize=(14,6))
ax = fig.add_subplot(1, 2, 1, projection='3d')
xg,yg = np.meshgrid(xy[:,0], xy[:,1])
ax.plot_wireframe(xy[:,0], xy[:,1], Hpsi)









    



[[ 1.14569979e-19  5.77897400e-19  2.75350811e-18 ... -3.00266964e-20
  -5.68754935e-21 -1.01004002e-21]]
(1, 3721)






    Out[4]:





<mpl_toolkits.mplot3d.art3d.Line3DCollection at 0x1e944ca14a8>



In [5]:

    
psiHpsi = lagar.psi_H_psi_integrand(np.transpose(xy),1)
print(psiHpsi)
print(psiHpsi.shape)
fig = plt.figure(figsize=(14,6))
ax = fig.add_subplot(1, 2, 1, projection='3d')
xg,yg = np.meshgrid(xy[:,0], xy[:,1])
ax.plot_wireframe(xy[:,0], xy[:,1], psiHpsi)









    



[[-3.82724579e-41 -1.00827354e-39 -2.37172564e-38 ... -2.68762858e-42
  -9.29558886e-44 -2.82985603e-45]]
(1, 3721)






    Out[5]:





<mpl_toolkits.mplot3d.art3d.Line3DCollection at 0x1e9376b9128>



In [6]:

    
psi_psi = lagar.psi_psi_integrand(np.transpose(xy),1)
print(psi_psi)
print(psi_psi.shape)
fig = plt.figure(figsize=(14,6))
ax = fig.add_subplot(1, 2, 1, projection='3d')
xg,yg = np.meshgrid(xy[:,0], xy[:,1])
ax.plot_wireframe(xy[:,0], xy[:,1], psi_psi)









    



[[1.11591482e-43 3.04407440e-42 7.41918458e-41 ... 8.01167634e-45
  2.67118186e-46 7.84967208e-48]]
(1, 3721)






    Out[6]:





<mpl_toolkits.mplot3d.art3d.Line3DCollection at 0x1e937577080>



In [ ]:



In [ ]:



In [ ]: