使用 tensorflow.estimator 训练 transformer 实现的问题

Question

以下colab url是使用Tensorflow简单实现Transformer算法

https://colab.research.google.com/drive/10mvd_0PECxAeOeUbTEI-6NWfdOr0tVQM?usp=sharing

在训练阶段，我收到以下错误信息。

**-------------------------------------------- ---------------------------- AttributeError Traceback（最后一次调用） 在() ----> 1 transformer.train(input_fn=lambda: train_input_fn(train_input_enc, train_output_dec, train_target_dec, batch_size), steps = epoch)

9帧 /usr/local/lib/python3.9/dist-packages/tensorflow/python/framework/ops.py 在 getattr(self, name) 第441话 第442章 --> 443 self.getattribute(名称) 444 第445话

AttributeError: 'Tensor' 对象没有属性 'value'

------------------------------------------------ --------------------------**

我花了好几个小时来弄清楚可能是什么问题，但我做不到。

请帮忙

我已经尝试彻底检查代码并在谷歌上搜索以找到可能的解决方案但没有成功。