當前位置: 首頁>>技術問答>>正文


在TensorFlow中如何將Tensor張量轉換為numpy數組?

在Tensorflow中,使用Python,如何將張量(Tensor)轉換為numpy數組呢?

tensorflow tensor

最佳解決辦法

Session.runeval返回的任何張量都是NumPy數組。

>>> print(type(tf.Session().run(tf.constant([1,2,3]))))
<class 'numpy.ndarray'>

要麽:

>>> sess = tf.InteractiveSession()
>>> print(type(tf.constant([1,2,3]).eval()))
<class 'numpy.ndarray'>

或者等同地:

>>> sess = tf.Session()
>>> with sess.as_default():
>>>    print(type(tf.constant([1,2,3]).eval()))
<class 'numpy.ndarray'>

numpy array數組

參考資料

本文由《純淨天空》出品。文章地址: https://vimsky.com/zh-tw/article/3725.html,未經允許,請勿轉載。