如何查看模型预测的时候每层的时延

我想知道模型每层的FLOPS是不是和推理时间成正相关的，请问有什么方法可以查看模型在推理的时候，每层的时延呢？