23. Tutorial de TensorFlow---Creación de gráficos

Una ecuación diferencial parcial (PDE) es una ecuación diferencial que involucra derivadas parciales de una función desconocida y múltiples variables independientes. En cuanto a las ecuaciones diferenciales parciales, nos centraremos en la creación de nuevas gráficas.

Supongamos que hay un estanque de tamaño 500*500 −

norte = 500

Ahora, calcularemos la ecuación diferencial parcial y la usaremos para formar la gráfica correspondiente. Considere los siguientes pasos para calcular una gráfica.

Paso 1 : importe la biblioteca utilizada para la simulación.

import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt


 

Paso 2 : incluye funciones para convertir matrices 2D en núcleos de convolución y operaciones de convolución 2D simplificadas.

def make_kernel(a):
   a = np.asarray(a)
   a = a.reshape(list(a.shape) + [1,1])
   return tf.constant(a, dtype=1)

def simple_conv(x, k):
   """一个简化的2D卷积操作"""
   x = tf.expand_dims(tf.expand_dims(x, 0), -1)
   y = tf.nn.depthwise_conv2d(x, k, [1, 1, 1, 1], padding = 'SAME')
   return y[0, :, :, 0]

def laplace(x):
   """计算数组的二维拉普拉斯"""
   laplace_k = make_kernel([[0.5, 1.0, 0.5], [1.0, -6., 1.0], [0.5, 1.0, 0.5]])
   return simple_conv(x, laplace_k)
   
se

Supongo que te gusta

Origin blog.csdn.net/Knowledgebase/article/details/133460037
Recomendado
Clasificación