TensorFlow 2.16 中 Keras Lambda 层出现异常“ValueError”

Question

我是 Tensorflow 新手，我按照几个教程来了解更多信息。 我的上一篇教程是关于变压器模型以及如何使用它们进行聊天的。 该模型在 TensorFlow 2.15 中编译良好，但在 TensorFlow 2.16 中出现异常。我不知道如何解决：

Exception has occurred: ValueError
Only input tensors may be passed as positional arguments. The following argument value should be passed as a keyword argument: 256 (of type <class 'int'>)

异常发生在lambda层：

  inputs = tf.keras.Input(shape = (None,), name = "inputs")
  padding_mask = tf.keras.Input(shape=(1, 1, None), name = "padding_mask")

  embeddings = tf.keras.layers.Embedding(vocabulary_size, model)(inputs)
  embeddings *= tf.keras.layers.Lambda(
      lambda model: tf.math.sqrt(tf.cast(model, tf.float32))
  )(model)

Answer 1

https://www.tensorflow.org/api_docs/python/tf/keras/layers/Embedding

描述 tf.keras.layers.Embedding(input_dim, output_dim, ...)

您将“model”设置为output_dim，这可能是一个错误。