簡體   English   中英

如何將 numpy 數組轉換為 keras 張量

[英]how to convert numpy array to keras tensor

使用keras模型進行預測時,出現以下錯誤

AttributeError: 'Tensor' 對象沒有屬性 'ndim'

原因是權重是 numpy 數組,而不是張量。
那么如何將 numpy 數組轉換為 keras 張量呢?

在 Tensorflow 中,它可以通過以下方式完成:

import tensorflow.keras.backend as K
import numpy as np

a = np.array([1,2,3])
b = K.constant(a)
print(b)

# <tf.Tensor 'Const_1:0' shape=(3,) dtype=float32>

print(K.eval(b))

# array([1., 2., 3.], dtype=float32)

在原始 keras 中,應該將import tensorflow.keras.backend as K替換import tensorflow.keras.backend as K from keras import backend as K

要將 numpy 數組轉換為張量,

import tensor as tf
#Considering y variable holds numpy array
y_tensor = tf.convert_to_tensor(y, dtype=tf.int64) 

#您可以使用任何最適合的可用數據類型 - https://www.tensorflow.org/api_docs/python/tf/dtypes/DType

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM