TensorFlow函數(shù):tf.nn.avg_pool

2019-01-31 11:29 更新

tf.nn.avg_pool函數(shù)

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

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

請參閱指南:神經(jīng)網(wǎng)絡(luò)>池操作

在輸入上執(zhí)行平均池.

output中的每個(gè)條目都是value中相應(yīng)大小ksize窗口的平均值.

參數(shù):

  • value:形狀為[batch, height, width, channels]的4-D Tensor,并且類型為float32,float64,qint8,quint8或qint32.
  • ksize:一個(gè)包含4個(gè)元素的1-D int Tensor,輸入張量的每個(gè)維度的窗口大小.
  • strides:一個(gè)包含4個(gè)元素的1-D int Tensor,輸入張量的每個(gè)維度的滑動窗口的步幅.
  • padding:一個(gè)字符串,可以是'VALID'或者'SAME'.填充算法.請參閱tf.nn.convolution
  • data_format:一個(gè)字符串,可以是'NHWC'和'NCHW'.
  • name:操作的可選名稱.

返回:

與value具有相同類型的Tensor.平均匯總輸出張量.

以上內(nèi)容是否對您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號