TensorFlow函數(shù)教程：tf.nn.dilation2d

tf.nn.dilation2d函數(shù)

tf.nn.dilation2d(
    input,
    filter,
    strides,
    rates,
    padding,
    name=None
)

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

請(qǐng)參閱指南：神經(jīng)網(wǎng)絡(luò)>形態(tài)學(xué)濾波

計(jì)算4-Dinput和3-Dfilter張量的灰度擴(kuò)張.

input張量具有shape[batch, in_height, in_width, depth],filter張量具有shape[filter_height, filter_width, depth],即,每個(gè)輸入通道都獨(dú)立于其他輸入通道進(jìn)行處理,具有自己的結(jié)構(gòu)函數(shù).該output張量具有shape[batch, out_height, out_width, depth].輸出張量的空間維度取決于padding算法.我們目前只支持默認(rèn)的“NHWC”data_format.

詳細(xì)地說(shuō),灰度形態(tài)2-D擴(kuò)張是最大和相關(guān)(為了與conv2d一致,我們使用未經(jīng)過(guò)鏡像的濾波器)：

output[b, y, x, c] =
   max_{dy, dx} input[b,
                      strides[1] * y + rates[1] * dy,
                      strides[2] * x + rates[2] * dx,
                      c] +
                filter[dy, dx, c]

當(dāng)過(guò)濾器的大小等于池內(nèi)核大小并包含全零時(shí),最大池是一種特殊情況.

二元性的注意事項(xiàng)：filter對(duì)input的擴(kuò)張等于反射filter對(duì)-input侵蝕的否定.

參數(shù)：

• input：一個(gè)4-DTensor,必須是下列類型之一：float32,float64,int32,uint8,int16,int8,int64,bfloat16,uint16,half,uint32,uint64,shape為[batch, in_height, in_width, depth].
• filter：一個(gè)3-DTensor,必須與input具有相同類型,shape為[filter_height, filter_width, depth].
• strides：ints列表,長(zhǎng)度>= 4.輸入張量的每個(gè)維度的滑動(dòng)窗口的步幅.必須是：[1, stride_height, stride_width, 1].
• rates：ints列表,長(zhǎng)度>= 4.輸入大步進(jìn)行劇烈的形態(tài)學(xué)擴(kuò)張.必須是：[1, rate_height, rate_width, 1].
• padding：string,可以是："SAME", "VALID".要使用的填充算法的類型.
• name：操作的名稱(可選).

返回：

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