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

2019-01-31 13:46 更新

tf.nn.depthwise_conv2d_native_backprop_input函數(shù)

tf.nn.depthwise_conv2d_native_backprop_input(
    input_sizes,
    filter,
    out_backprop,
    strides,
    padding,
    data_format='NHWC',
    dilations=[1, 1, 1, 1],
    name=None
)

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

請(qǐng)參閱指南:神經(jīng)網(wǎng)絡(luò)>卷積運(yùn)算

計(jì)算相對(duì)于輸入的深度卷積的梯度.

參數(shù):

  • input_sizes:一個(gè)int32類型的Tensor.一個(gè)整數(shù)向量,表示input的shape,它是基于data_format的.例如,如果data_format是'NHWC',則input是 4-D [batch, height, width, channels]張量.
  • filter:一個(gè)4-DTensor,必須是下列類型之一:half,bfloat16,float32,float64, shape為[filter_height, filter_width, in_channels, depthwise_multiplier].
  • out_backprop:一個(gè)4-DTensor,必須與filter具有相同類型,shape是基于data_format的.例如,如果data_format是'NHWC',那么out_backprop的shape是[batch, out_height, out_width, out_channels].梯度與卷積的輸出相關(guān).
  • strides:列表ints的列表,對(duì)于卷積輸入的每個(gè)維度,滑動(dòng)窗口的步幅.
  • paddingstring,可以是:"SAME", "VALID".要使用的填充算法的類型.
  • data_format:可選的string可以是:"NHWC", "NCHW",默認(rèn)為"NHWC".指定輸入和輸出數(shù)據(jù)的數(shù)據(jù)格式.使用默認(rèn)格式“NHWC”,數(shù)據(jù)按以下順序存儲(chǔ):[batch, height, width, channels].或者,格式可以是“NCHW”,數(shù)據(jù)存儲(chǔ)順序?yàn)椋?span>[batch, channels, height, width].
  • dilationsints的可選列表,默認(rèn)為[1, 1, 1, 1].長(zhǎng)度為4的1-D張量.input每個(gè)維度的擴(kuò)張系數(shù).如果設(shè)置為k> 1,則該維度上的每個(gè)過(guò)濾器元素之間將有k-1個(gè)跳過(guò)的單元格.維度順序由值data_format確定.批次和深度維度的擴(kuò)張必須為1.
  • name:操作的名稱(可選).

返回:

一個(gè)Tensor,filter具有相同的類型.

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

掃描二維碼

下載編程獅App

公眾號(hào)
微信公眾號(hào)

編程獅公眾號(hào)