阅读(8.9k) 书签 (0)

TensorFlow函数:tf.nn.avg_pool

2019-01-31 11:29 更新

tf.nn.avg_pool函数

tf.nn.avg_pool(
    value,
    ksize,
    strides,
    padding,
    data_format='NHWC',
    name=None
)

定义于tensorflow/python/ops/nn_ops.py.

请参阅指南:神经网络>池操作

在输入上执行平均池.

output中的每个条目都是value中相应大小ksize窗口的平均值.

参数:

  • value:形状为[batch, height, width, channels]的4-D Tensor,并且类型为float32,float64,qint8,quint8或qint32.
  • ksize:一个包含4个元素的1-D int Tensor,输入张量的每个维度的窗口大小.
  • strides:一个包含4个元素的1-D int Tensor,输入张量的每个维度的滑动窗口的步幅.
  • padding:一个字符串,可以是'VALID'或者'SAME'.填充算法.请参阅tf.nn.convolution
  • data_format:一个字符串,可以是'NHWC'和'NCHW'.
  • name:操作的可选名称.

返回:

与value具有相同类型的Tensor.平均汇总输出张量.