這篇文章主要講解了在keras中如何實現(xiàn)獲取張量tensor的維度大小,內容清晰明了,對此有興趣的小伙伴可以學習一下,相信大家閱讀完之后會有幫助。

在進行keras 網(wǎng)絡計算時,有時候需要獲取輸入張量的維度來定義自己的層。但是由于keras是一個封閉的接口。因此在調用由于是張量不能直接用numpy 里的A.shape()。這樣的形式來獲取。這里需要調用一下keras 作為后端的方式來獲取。當我們想要操作時第一時間就想到直接用 shape ()函數(shù)。其實keras 中真的有shape()這個函數(shù)。
shape(x)返回一個張量的符號shape,符號shape的意思是返回值本身也是一個tensor,
示例:
>>> from keras import backend as K >>> tf_session = K.get_session() >>> val = np.array([[1, 2], [3, 4]]) >>> kvar = K.variable(value=val) >>> input = keras.backend.placeholder(shape=(2, 4, 5)) >>> K.shape(kvar) <tf.Tensor 'Shape_8:0' shape=(2,) dtype=int32> >>> K.shape(input) <tf.Tensor 'Shape_9:0' shape=(3,) dtype=int32> __To get integer shape (Instead, you can use K.int_shape(x))__ >>> K.shape(kvar).eval(session=tf_session) array([2, 2], dtype=int32) >>> K.shape(input).eval(session=tf_session) array([2, 4, 5], dtype=int32)
文章標題:在keras中如何實現(xiàn)獲取張量tensor的維度大小-創(chuàng)新互聯(lián)
文章分享:http://chinadenli.net/article8/cdcdip.html
成都網(wǎng)站建設公司_創(chuàng)新互聯(lián),為您提供網(wǎng)站策劃、面包屑導航、定制網(wǎng)站、網(wǎng)站維護、企業(yè)網(wǎng)站制作、網(wǎng)站導航
聲明:本網(wǎng)站發(fā)布的內容(圖片、視頻和文字)以用戶投稿、用戶轉載內容為主,如果涉及侵權請盡快告知,我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場,如需處理請聯(lián)系客服。電話:028-86922220;郵箱:631063699@qq.com。內容未經(jīng)允許不得轉載,或轉載時需注明來源: 創(chuàng)新互聯(lián)