ValueError:Tensor转换请求使用dtype float32 LSTM实现的Tensor的dtype int32(tensorflow 2.0.0)

2024-04-24 10:56:17 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图测试LSTM的不同类型的实现,并在预测时在代码中遇到这个错误

Tensorflow版本-“2.0.0”


我并没有删除这个问题,因为我仍然需要知道哪里出了问题。在输入模型时,我是否总是需要担心将float32作为数据类型


示例代码

X = list()
Y = list()
X = [x+1 for x in range(20)]
Y = [y * 15 for y in X]

X = np.array(X,dtype=int)
Y=  np.array(Y,dtype=int)

X=array(X).reshape(20, 1, 1)


model = Sequential()
model.add(LSTM(50, activation='relu', input_shape=(1, 1)))
model.add(Dense(1))
model.compile(optimizer='adam', loss='mse')
print(model.summary())

model.fit(X, Y, epochs=2, validation_split=0.2, batch_size=5)


test_input = np.array(30,dtype=int)
test_input = test_input.reshape((1, 1, 1))
test_output = model.predict(test_input)   <---- ERROR IN THIS LINE

错误:

ValueError Traceback (most recent call last)
~\AppData\Local\Continuum\anaconda3\envs\PythonCPU\lib\site-packages\tensorflow_core\python\framework\op_def_library.py in _apply_op_helper(self, op_type_name, name, **keywords)
    526                 as_ref=input_arg.is_ref,
--> 527                 preferred_dtype=default_dtype)
    528           except TypeError as err:

~\AppData\Local\Continuum\anaconda3\envs\PythonCPU\lib\site-packages\tensorflow_core\python\framework\ops.py in internal_convert_to_tensor(value, dtype, name, as_ref, preferred_dtype, ctx, accept_composite_tensors)
   1270           "Tensor conversion requested dtype %s for Tensor with dtype %s: %r" %
-> 1271           (dtype.name, value.dtype.name, value))
   1272     return value

ValueError: Tensor conversion requested dtype int32 for Tensor with dtype float32: <tf.Tensor 'sequential/lstm/MatMul/ReadVariableOp:0' shape=(1, 200) dtype=float32>


During handling of the above exception, another exception occurred:
.....
.....
.....
TypeError: Input 'b' of 'MatMul' Op has type float32 that does not match type int32 of argument 'a'.

Tags: nameintestforinputmodelvaluetype
1条回答
网友
1楼 · 发布于 2024-04-24 10:56:17

我尝试了不同的数据类型排列,结果发现我只需将所有数组中的所有数据类型更改为“float32”即可解决错误

相关问题 更多 >