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

2019-01-31 13:44 更新

tf.nn.conv2d_backprop_input函數(shù)

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

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

計算相對于輸入的卷積梯度.

參數(shù):

  • input_sizes:一個int32類型的Tensor,表示input形狀的整數(shù)向量,其中input是4-D [batch, height, width, channels]張量.
  • filter:一個Tensor,必須是下列類型之一:half,bfloat16,float32,float64,是4-D維的,并且形狀為[filter_height, filter_width, in_channels, out_channels].
  • out_backprop:一個Tensor,必須與filter相同,是4-D的,且形狀為[batch, out_height, out_width, out_channels],梯度與卷積的輸出相關(guān).
  • strides:ints列表,對于卷積輸入的每個維度,滑動窗口的步幅,必須與使用format指定的維度處于相同的順序.
  • padding:string,可以是:"SAME", "VALID",要使用的填充算法的類型.
  • use_cudnn_on_gpu:bool,默認(rèn)為True.
  • data_format:string,可以是:"NHWC", "NCHW".默認(rèn)為"NHWC",指定輸入和輸出數(shù)據(jù)的數(shù)據(jù)格式,使用默認(rèn)格式“NHWC”,數(shù)據(jù)按以下順序存儲:[batch,in_height,in_width,in_channels].或者,格式可以是“NCHW”,數(shù)據(jù)存儲順序為:[batch,in_channels,in_height,in_width].
  • dilations:可選列表ints,默認(rèn)為[1, 1, 1, 1],長度為4的1-D張量,input每個維度的擴張系數(shù);如果設(shè)置為k> 1,則該維度上的每個濾鏡元素之間將有k-1個跳過的單元格;維度順序由data_format值確定,詳見上文;批次和深度尺寸的擴張必須為1.
  • name:操作的名稱(可選).

返回:

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

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

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號