Tensorflow 基本用法（二）

最新推荐文章于 2025-05-23 12:18:51 发布

原创最新推荐文章于 2025-05-23 12:18:51 发布 · 785 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#Tensorflow #图像处理 #神经网络

Tensorflow 专栏收录该内容

2 篇文章

订阅专栏

本文深入介绍了TensorFlow中常用的六个函数，包括随机初始化张量、创建常数张量、二维卷积、最大池化、ReLU激活函数及Dropout层的具体用法与参数说明，并通过实例演示了这些函数的应用。

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

介绍在官方文档中MNIST进阶中用到的函数

1. tf.truncated_normal(shape, mean=0.0, stddev=1.0, dtype=tf.float32, seed=None, name=None)

shape：生成张量的形状
mean：正态分布的均值
stddev：正太分布的标准差
dtype：数据类型
seed：种子
name：操作的命名（可以不指定）
返回值是随机截断的正太分布值填充的张量

initial = tf.truncated_normal([3,3], stddev=0.1)
sess = tf.InteractiveSession()
print(initial.eval())
sess.close()
#[[-0.1338151  -0.01149522 -0.00680522]
  [ 0.07212916 -0.1230556   0.07994577]
  [ 0.1255837  -0.1215869  -0.13001645]]

2. tf.constant(value, dtype=None, shape=None, name=’Const’)

value：用于填充的数值或列表
dtype：数据类型
shape：生成张量的形状
name：操作的命名（可以不指定）
返回值一个常数张量

# Constant 1-D Tensor populated with value list.
tensor = tf.constant([1, 2, 3, 4, 5, 6, 7]) => [1 2 3 4 5 6 7]

# Constant 2-D tensor populated with scalar value -1.
tensor = tf.constant(-1.0, shape=[2, 3]) => [[-1. -1. -1.]
                                             [-1. -1. -1.]]
# Constant 2-D tensor populated with value list.
tensor = tf.constant([1,2,3,4,5,6], shape=[2,3]) => [[1 2 3]
                                                     [4 5 6]]

3. tf.nn.conv2d(input, filter, strides, padding, use_cudnn_on_gpu=True, data_format=’NHWC’, dilations=[1, 1, 1, 1], name=None)

input：输入张量，形状是4D
filter：卷积核，形状是4D张量，[核高度 , 核宽度 , 输入通道数, 输出通道数]
strides：步长
padding：填充方式，只能选SAME或VALID，SAME在图像边缘进行0填充，VALID不进行填充
use_cudnn_on_gpu：使用gpu加速
data_format：输入维度格式，NHWC依次是批，高度，宽度，通道；NCHW依次是批，通道，高度，宽度
dilations：输入的扩张
name：操作的命名（可以不指定）
返回值一个类型与输入相同的张量

具体操作在这篇博客介绍的非常详细 http://www.cnblogs.com/welhzh/p/6607581.html

4. tf.nn.max_pool(value, ksize, strides, padding, data_format=’NHWC’, name=None)

value：输入张量，形状是4D，格式由data_format
ksize：输入张量每个维度的窗口大小，一个4个元素的列表或元组
strides：步长
padding：填充方式，只能选SAME或VALID，SAME在图像边缘进行0填充，VALID不进行填充
data_format：输入维度格式，NHWC依次是批，高度，宽度，通道；NCHW依次是批，通道，高度，宽度
name：操作的命名（可以不指定）
返回值是经过最大池化的data_format格式的张量

sess = tf.InteractiveSession()
a = tf.constant(np.arange(9),shape = [1,3,3,1])
print(a.eval())
'''
 [[[[0]
   [1]
   [2]]
  [[3]
   [4]
   [5]]
  [[6]
   [7]
   [8]]]]
'''
print(a.eval)
#<bound method Tensor.eval of <tf.Tensor 'Const_1:0' shape=(1, 3, 3, 1) dtype=int32>>
b = tf.nn.max_pool(a,ksize=[1, 2, 2, 1],strides=[1, 2, 2, 1], padding='SAME')
print(b.eval())
'''
[[[[4]
   [5]]
  [[7]
   [8]]]]
'''
print(b.eval)
#<bound method Tensor.eval of <tf.Tensor 'MaxPool:0' shape=(1, 2, 2, 1) dtype=int32>> 形状改变了
sess.close()

5. tf.nn.relu(features, name=None)

features：输入张量（图像特征）
name：操作的命名（可以不指定）
返回值与输入类型相同的张量

sess = tf.InteractiveSession()
a = tf.constant([-1,-2,2,1],shape = [4])
print(a.eval())
#[-1 -2 2 1]
b = tf.nn.relu(a)
print(b.eval())
#[0 0 2 1]
sess.close()

6. tf.nn.dropout(x, keep_prob, noise_shape=None, seed=None, name=None)

x：输入张量
keep_prob：神经元被选中概率
noise_shape：选中/丢弃
seed：种子
name：操作的命名（可以不指定）
返回值是一个与输入相同形状的张量

keep_prob = tf.placeholder("float")
h_fc1_drop = tf.nn.dropout(h_fc1, keep_prob)#必须设置keep_prob，并且keep_prob是一个占位符