InvalidArgumentError Traceback most recent call lastCell In12 line 15 11 for i in rangeiter + 1 13 with tfGradientTape as tape--- 15 PRED_train = tfmatmulX_tra

出错的原因是输入的张量类型不匹配。根据错误信息，输入的张量X_train和W的类型应该是float类型，但实际上它们的类型是double。

要解决这个问题，可以将输入的张量类型转换为float类型。可以使用tf.cast()函数来实现类型转换。修改代码如下：

X_train = tf.cast(X_train, tf.float32)
W = tf.cast(W, tf.float32)

在计算tf.matmul()之前，将X_train和W转换为float类型。这样就可以解决InvalidArgumentError错误。