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

2019-01-31 11:29 更新

tf.nn.avg_pool3d函數(shù)

tf.nn.avg_pool3d(
    input,
    ksize,
    strides,
    padding,
    data_format='NDHWC',
    name=None
)

定義在:tensorflow/python/ops/gen_nn_ops.py.

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

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

參數(shù):

  • input:一個Tensor,并且必須是下列類型之一:half,bfloat16,float32,float64.形狀[batch, depth, rows, cols, channels]張量到池.
  • ksize:ints列表中有長度>= 5,長度為5的1-D張量,輸入張量的每個維度的窗口大小,必須有:ksize[0] = ksize[4] = 1.
  • strides:ints列表中有長度>= 5,長度為5的1-D張量,input中每個維度的滑動窗口的步幅.必須有:strides[0] = strides[4] = 1.
  • padding:來自:"SAME", "VALID"的一個string.要使用的填充算法的類型.
  • data_format:來自:"NDHWC", "NCDHW"的可選string,默認為"NDHWC",輸入和輸出數(shù)據(jù)的數(shù)據(jù)格式,使用默認格式“NDHWC”,數(shù)據(jù)按以下順序存儲:[batch,in_depth,in_height,in_width,in_channels];或者,格式可以是“NCDHW”,數(shù)據(jù)存儲順序是:[batch,in_channels,in_depth,in_height,in_width].
  • name:操作的名稱(可選).

返回:

與input具有相同的類型的Tensor.

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

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號