六、tensorflow之添加层。

最新推荐文章于 2020-04-14 20:00:54 发布

paranoid_CNN

最新推荐文章于 2020-04-14 20:00:54 发布

阅读量591

点赞数

CC 4.0 BY-SA版权

分类专栏： tensorflow学习

本文链接：https://blog.youkuaiyun.com/paranoid_CNN/article/details/78289324

tensorflow学习专栏收录该内容

8 篇文章

订阅专栏

本文介绍如何使用Python在TensorFlow中定义层。通过一个具体例子，展示了如何创建一个包含权重和偏置的层，并实现了激活函数的加入。此过程有助于理解神经网络的基本组件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

对于tensorflow来讲，利用python添加层是通过定义函数定义的。
比如，定一个层，实现activation(x*w+b) , activation是激活函数。

import tensorflow as tf
def add_layer(inputs, in_size, out_size, activation=None):
  Weights = tf.Variable(tf.random_normal([in_size,
  out_size]))
  biases = tf.Variable(tf.zeros([1, out_size]))+0.1
  Wx_plus_b = tf.matmul(inputs, Weights) + biases
  if activation = None:
    outputs=Wx_plus_b
  else:
    outputs= activation( Wx_plus_b)
  return outputs