在tensorflow[1.x/2.x]下,實現tensor與numpy互轉

在tensorflow的開發中,常常需要將tensor與numpy互相配合,而是實現特定的功能。而tensor與numpy的互相轉換,必不可少。

請注意,tf2因爲使用eager機制,轉換時不需要new session。出現如下錯誤,多半是沒有搞清楚所在環境。‘Tensor’ object has no attribute ‘numpy’

TF1.x

tensor -> numpy

    with tf.Session() as sess:
        numpy_data = tensor_data.eval()

numpy->tensor

tensor_data= tf.convert_to_tensor(numpy_data)

TF2.x

tensor -> numpy

numpy_data = tensor_data.numpy()

numpy -> tensor

tensor_data = tf.cast(numpy_data, dtype=tf.float32)#numpy轉張量
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章