TensorFlow常用函数

来源:互联网 发布:生产环境 调试java 编辑:程序博客网 时间:2024/06/11 17:33

1. tf.nn.conv2d(input, filter, strides, padding, use_cudnn_on_gpu = None, data_format = None, name = None)

计算张量的2-D卷积

  • input: 4-D张量。[batch, in_height, in_width, in_channels]
  • filter: 与input有相同类型。[filter_height, filter_width, in_channels, out_channels]
  • strides: 一个长度是4的一维整数类型数组,每一维度对应的是input中每一维的对应移动步数
  • padding: 可选’SAME’或’VALID’,前者表示仅适用于全尺寸操作(补全的方式),即输入数据维度和输出数据维度相同;后者适用于部分窗口(丢弃的方式),即输入数据维度和输出数据维度不同,具体为[(input-filter+1)/strides]取较大的整数,比如28*28经过11*11卷积得到18*18的结果
  • 输出:一个具有与input相同类型的张量

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

最大池化操作

  • value: 4维的tf.float32类型的张量。[batch, height, width, channels]
  • ksize: 长度≥4的整型列表,输入每个维度的窗口大小
  • strides: 长度≥4的整型列表,输入每个维度窗口的滑动步数
  • padding: ‘SAME”VALID’,同上
  • data_format: ‘NHWC’,’NCHW’
  • name: 操作的可选名称,可不填
  • 返回:tf.float32类型张量

3. tf.transpose()

张量转置操作

后续持续更新

原创粉丝点击