使用tf.fucntion时如何为函数名称和模型名称编写Input_Signature？

Question

我想将tf.function装饰为我的训练功能

@tf.function
def trainOneSample(data, model, optimizer, lossFunc, accFunc):
...

其中数据是一对具有float32和uint8类型（用于输入和标签）的5D数组，model是我的NN模型的名称，optimiser是优化对象的名称，lossFunc和accFunc是自定义函数的名称。如何通过Input_Signature在tf.function中装饰它们？我无法在tf的教程或指南中找到任何解决方案。