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

2019-01-31 13:44 更新

tf.nn.conv2d_backprop_filter函數(shù)

tf.nn.conv2d_backprop_filter(
    input,
    filter_sizes,
    out_backprop,
    strides,
    padding,
    use_cudnn_on_gpu=True,
    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:一個(gè)Tensor,必須是下列類型之一:half,bfloat16,float32,float64,是4-D的,并且具有形狀[batch, in_height, in_width, in_channels].
  • filter_sizes:一個(gè)Tensor,類型為int32,表示filter張量形狀的整數(shù)向量,其中filter是4-D [filter_height, filter_width, in_channels, out_channels]張量.
  • out_backprop:一個(gè)Tensor,必須與input相同;4-D的,形狀為[batch, out_height, out_width, out_channels],梯度與卷積的輸出相關(guān).
  • strides:ints列表,對(duì)于卷積輸入的每個(gè)維度,滑動(dòng)窗口的步幅.必須與使用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ù)按以下順序存儲(chǔ):[batch,in_height,in_width,in_channels].或者,格式可以是“NCHW”,數(shù)據(jù)存儲(chǔ)順序?yàn)椋篬batch,in_channels,in_height,in_width].
  • dilations:ints的可選列表,默認(rèn)為[1, 1, 1, 1],長(zhǎng)度為4的1-D張量,input每個(gè)維度的擴(kuò)張系數(shù),如果設(shè)置為k> 1,則該維度上的每個(gè)濾鏡元素之間將有k-1個(gè)跳過(guò)的單元格.維度順序由值data_format確定,詳見上文.批次和深度維度的擴(kuò)張必須為1.
  • name:操作的名稱(可選).

返回:

一個(gè)Tensor,與input有相同的類型.

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

掃描二維碼

下載編程獅App

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

編程獅公眾號(hào)