TensorFlow函數(shù)教程:tf.nn.max_pool3d

2019-01-31 13:50 更新

tf.nn.max_pool3d函數(shù)

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

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

在輸入上執(zhí)行3D最大池化.

參數(shù):

  • input:一個Tensor.必須是下列類型之一:half,bfloat16,float32.shape為[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:string,可以是:"SAME", "VALID".要使用的填充算法的類型.
  • data_format:可選的string,可以是:"NDHWC", "NCDHW".默認為"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:操作的名稱(可選).

返回:

一個Tensor,與input有相同的類型.

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

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號