tensorlfow數據類型之reference type

a=tf.Variable(1,name=“variable_a”)
print(a.dtype)
在python層面上述代碼產生一個int32_ref的tensor,喫a的op必須和variable_a在同一設備上,並且tensor a 和variable_a共享同一塊buffer(共享buffer肯定要在同一設備上)。上述代碼產生的pb圖如下:
在這裏插入圖片描述
我們看看read的屬性:
在這裏插入圖片描述
注意_class屬性,loc:@Variable表示此op的輸出tensor和Variable這個op共享buffer,也必須在同一設備上。
再從c++底層解析:
看tensorflow-r1.14\tensorflow\core\ops\state_ops.cc
在這裏插入圖片描述
註冊Variable op的時候,就限制其輸出爲Ref。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章