自定义层tensorflow python中当前历元的编号

这是我第一次问问题，所以对我放松点。我正在尝试用python tensorflow/keras编写一个自定义层，并在该层中访问模型当前历元的编号。我已经广泛地搜索了tensorflow文档，以找到一些可以给我这个信息的东西。我能找到的最好的方法是

tensorflow.keras.backend.learning\u phase（）

，它返回是否是学习阶段。请帮忙

您需要创建回调，请检查

并确保这是所谓的某个地方：

model.fit(callbacks=[CustomEpoch()])

model.fit(callbacks=[CustomEpoch()])