Saya membuat jaringan saraf dengan Tensorflow
dan skflow
; untuk beberapa alasan saya ingin mendapatkan nilai dari beberapa tensor dalam untuk input tertentu, jadi saya menggunakan myClassifier.get_layer_value(input, "tensorName")
, myClassifier
menjadi a skflow.estimators.TensorFlowEstimator
.
Namun, saya merasa sulit untuk menemukan sintaks yang benar dari nama tensor, bahkan mengetahui namanya (dan saya bingung antara operasi dan tensor), jadi saya menggunakan tensorboard untuk menggambar grafik dan mencari namanya.
Apakah ada cara untuk menghitung semua tensor dalam grafik tanpa menggunakan tensorboard?
sumber
if "Variable" in n.op
di akhir pemahaman.Ada cara untuk melakukannya sedikit lebih cepat daripada jawaban Yaroslav dengan menggunakan get_operations . Berikut ini contoh singkatnya:
sumber
tf.get_operations()
. Hanya operasi yang bisa Anda dapatkan.Saya akan mencoba meringkas jawabannya:
Untuk mendapatkan semua node (tipe
tensorflow.core.framework.node_def_pb2.NodeDef
):Untuk mendapatkan semua operasi (tipe
tensorflow.python.framework.ops.Operation
):Untuk mendapatkan semua variabel (tipe
tensorflow.python.ops.resource_variable_ops.ResourceVariable
):Untuk mendapatkan semua tensor (tipe
tensorflow.python.framework.ops.Tensor
) :sumber
tf.all_variables()
dapat memberi Anda informasi yang Anda inginkan.Selain itu, commit ini dibuat hari ini di TensorFlow Learn yang menyediakan fungsi
get_variable_names
di estimator yang bisa Anda gunakan untuk mengambil semua nama variabel dengan mudah.sumber
tf.global_variables()
all_variables (from tensorflow.python.ops.variables) is deprecated and will be removed after 2017-03-02
Saya pikir ini juga akan berhasil:
Tetapi dibandingkan dengan jawaban Salvado dan Yaroslav, saya tidak tahu mana yang lebih baik.
sumber
Jawaban yang diterima hanya memberi Anda daftar string dengan nama. Saya lebih suka pendekatan yang berbeda, yang memberi Anda (hampir) akses langsung ke tensor:
list_of_tuples
sekarang berisi setiap tensor, masing-masing di dalam tupel. Anda juga bisa menyesuaikannya untuk mendapatkan tensor secara langsung:sumber
Karena OP meminta daftar tensor dan bukan daftar operasi / node, kodenya harus sedikit berbeda:
sumber
Jawaban sebelumnya bagus, saya hanya ingin membagikan fungsi utilitas yang saya tulis untuk memilih Tensor dari grafik:
Jadi jika Anda memiliki grafik dengan ops:
Lalu lari
kembali:
sumber
Ini berhasil untuk saya:
sumber