欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

tensorflow学习笔记(北京大学) tf4_8_forward.py 完全解析

程序员文章站 2022-07-13 12:22:24
...
#coding:utf-8
#**tensorflow学习笔记(北京大学) tf4_8_forward.py 完全解析**  
#QQ群:476842922(欢迎加群讨论学习)
#如有错误还望留言指正,谢谢
#0导入模块 ,生成模拟数据集
import tensorflow as tf

#定义神经网络的输入、参数和输出,定义前向传播过程 
def get_weight(shape, regularizer):#权重
	w = tf.Variable(tf.random_normal(shape), dtype=tf.float32)#正态分布权重
	tf.add_to_collection('losses', tf.contrib.layers.l2_regularizer(regularizer)(w))
	return w
#tf.add_to_collection(‘list_name’, element):将元素element添加到列表list_name中

def get_bias(shape):  #偏执b
    b = tf.Variable(tf.constant(0.01, shape=shape)) #偏执b=0.01
    return b
	
def forward(x, regularizer):#前向传播
	
	w1 = get_weight([2,11], regularizer)#设置权重	
	b1 = get_bias([11])#设置偏执
	y1 = tf.nn.relu(tf.matmul(x, w1) + b1)#计算图

	w2 = get_weight([11,1], regularizer)#设置权重
	b2 = get_bias([1])#设置偏执
	y = tf.matmul(y1, w2) + b2 #计算图
	
	return y