Tensorflow 獲取Tensor的維度

2021-08-16 22:35:22 字數 485 閱讀 1313

問題:

typeerror: fetch argument dimension(1152) has invalid type , must be a string or tensor. (can not convert a dimension into a tensor or operation.)

tensorflow 中獲取tensor 的 shape 有兩種方法:tensor.shape, tensor.get_shape,兩種方法獲得的都是tensor變數,如果想要得到shape列表,可以通過一下方式獲取:

tensor.shape.as_list()       # out: [2, 3]

tensor.get_shape().as_list()    #out: [2, 3]

還可以通過一下方法獲取維度的個數:

tensor.shape.ndims  # out: 2

tensor.get_shape().ndims # out: 2

CentOS 通過原始碼編譯形式安裝 Tengine

本篇簡單說說在如何在 linux 系統上通過原始碼編譯的形式安裝 tengine,純屬筆記。tengine 官網路徑 linux 版本 centos6.9 x64 tengine 版本 tengine 2.3.2.tar.gz 關閉 linux 機器防火牆 上傳 tengine 安裝包 tengin...

TensorFlow在特定的作用域下獲取變數

一 例項描述 在作用域下,可以使用get variable和巢狀variable scope。get variable建立兩個同樣名字的變數是行不通的,如下 會報錯。var1 tf.get variable firstvar shape 2 dtype tf.float32 var2 tf.get ...

TensorFlow 簡易安裝

首先要安裝python,這個就不多說了,網上大把教程。註明一下我是使用的是linux系統。接下來是具體tensorflow的安裝過程了。tensorflow可以看做是乙個用來解決問題的工具包。對於工具包的安裝就有不同的途徑,1 通過一些python的包管理系統來進行安裝 2 通過原始碼自行安裝。對於...