越來(lái)越感覺(jué)學(xué)習(xí)python一個(gè)第三方庫(kù)就跟學(xué)一門(mén)新語(yǔ)言差不多,龐大的TensorFlow更是如此。這篇博客會(huì)持續(xù)記錄關(guān)于tensorflow的一些api的用法。
在tensorflow中,最基本的變量或者常量都是tensor類(lèi)型,說(shuō)直接點(diǎn)就是多維數(shù)組,因此涉及到了很多在某一個(gè)維度上操作的API,比如tf.reduce_*()
,tf_arg*()
,tf_expand_dims()
等,這些基本的API都有一個(gè)共同的參數(shù):axis=
,也就說(shuō)這些操作都是哪些維度(軸)上做運(yùn)算,除了在tf中,在numpy里面也有這一參數(shù),之前一直是一知半解?,F(xiàn)在總結(jié)一下這個(gè)axis,由于numpy與tensorflow的axis基本一樣,因此下面以numpy為例。
首先可以從多維數(shù)組(tensor)的shape來(lái)得到數(shù)組的維度: d=len(array.shape)
:
|
|
也就是說(shuō)上述的數(shù)組a的shape為(2,3)
,維度是2維,也就是有兩層嵌套數(shù)組,看看有多少層[]
,因此axis
的取值只能為0,1
,下面以sum
為例,看看不同的axis的結(jié)果:
|
|
一個(gè)一個(gè)的說(shuō)明。首先需要明白操作的對(duì)象是誰(shuí)?這這個(gè)例子中,就是那些元素相加。怎么來(lái)確定是哪些元素呢?個(gè)人使用的原則是:
對(duì)原來(lái)數(shù)組去掉axis+1層嵌套([]),對(duì)剩下的元素再分組進(jìn)行相應(yīng)的操作。
首先如果不加axis
參數(shù),則默認(rèn)對(duì)所有元素直接做sum
操作,因此結(jié)果就是21。當(dāng)axis=0
的時(shí)候, 相加的對(duì)象,則是變成了[1,2,3]+[4,5,6]
,按照上面的原則,a去掉一層嵌套,也就是[]
之后,變?yōu)? [1,2,3],[4,5,6]
那么sum
的對(duì)象: [1,2,3]+[4,5,6]=[5,7,9]
。 再接著看axis=1
,這時(shí)候需要去掉兩層嵌套,去掉第一層變?yōu)?[1,2,3], [4,5,6]
,去掉第二層嵌套,剩下了1,2,3
和 4,5,6
兩個(gè)組,因此兩個(gè)組內(nèi)各個(gè)相加即可,得到的最終的結(jié)果為:[6,15]
。 下面以同樣的思路,計(jì)算最小值:
|
|
有一個(gè)稍微特殊的是expand_dims
這個(gè)是擴(kuò)展維度。對(duì)于這個(gè)api它的axis沒(méi)有上限約束,不過(guò)當(dāng)所有的元素自成一個(gè)列表之后,不再變化。它的axis
參數(shù)不一樣的地方,是在axis
的位置加入新的一個(gè)軸,也就是說(shuō)此時(shí)對(duì)原數(shù)組去掉axis
層嵌套,而非axis+1
層,然后再對(duì)每組進(jìn)行加一個(gè)軸,看例子:
|
|
到現(xiàn)在基本上tensorflow以及numpy的axis這個(gè)參數(shù)基本沒(méi)有問(wèn)題了。 下面再看幾個(gè)常用的函數(shù)。
這個(gè)函數(shù)主要用于對(duì)多維數(shù)組的截取(切片),首先看看原型:
|
|
直接看文檔的例子:
|
|
先看第一個(gè)例子:
begin=[1,0,0]
也就是開(kāi)始的元素,這里就是input[1][0][0] = 3
,就是第二行[3,3,3]
的第一個(gè)3,,再看截取距離: [1,1,3]
,首先在axis=0
的上截取一個(gè)距離,得到是第二整行[[3,3,3],[4,4,4]]
,在axis=1
上截取2個(gè)距離,得到[3,3,3]
,最后在axis=2
上截取了3個(gè)距離,也就是[3,3,3]
這三個(gè)元素都得到,最終結(jié)果為:[3,3,3]
。
第二個(gè)例子解析:
開(kāi)始的元素仍然是input[1][0][0=0]
,在axis=0
上截取1個(gè)距離,得到仍然是[[[3,3,3],[4,4,4]]]
,第二個(gè)維度上截取距離為2,得到[[[3,3,3],[4,4,4]]]
,最后在第三個(gè)維度上截取3,即三個(gè)元素都保留,得到結(jié)果。這里如果第三個(gè)維度截取2的話(huà),那么得到就是[[[3,3],[4,4]]]
。
第三個(gè)同理,不再敘述。
參考: 1
聯(lián)系客服