TensorFlow中层API：Layers

TensorFlow中层API：Layers

Layers模块属于TensorFlow的一个稳定的中层API，其源码位于tensorflow/python/layers/layers.py，其官方文档地址为https://www.tensorflow.org/api_docs/python/tf/layers ref1

tf.layers这个中层API基本上可以算是tf.nn模块的抽象，可以极大地加快模型的构建速度。

tf.layers里面有很多封装好的类和函数。

类：

密集连接类

• Dense

Dropout类

• Dropout

卷积类

• Conv1D
• Conv2D
• Conv3D

BN类

• BatchNormalization

解卷积类（卷积转置）

• Conv2DTranspose
• Conv3DTranspose

平均池化类

• AveragePooling1D
• AveragePooling2D (e.g. images)
• AveragePooling3D (e.g. volumes)

最大池化类

• MaxPooling1D
• MaxPooling2D
• MaxPooling3D

Flatten类

• Flatten

深度可分离卷积

• SeparableConv1D 一维深度可分离卷积层
• SeparableConv2D 两维深度可分离卷积层

其他类

• Input
• Layer

函数：

这里的函数其实是基于上面的类实现的

• average_pooling1d
• average_pooling2d
• average_pooling3d
• batch_normalization
• conv1d
• conv2d
• conv2d_transpose
• conv3d
• conv3d_transpose
• dense
• dropout
• flatten
• max_pooling1d
• max_pooling2d
• max_pooling3d
• separable_conv1d
• separable_conv2d

类详解

1.Input

tf.layers.Input用于输入数据的类，有点类似于tf.plcaeholder，相当于一个占位符作用

#用法
Input = tf.layers.Input(dtype=tf.float32, shape=[None, 224, 224, 3])

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！