请教一下加速方案

我在预测时发现对未来的天预测时，是循环天数，每一次都重新计算一遍，batch_size=512,sample_count=5，一次计算需要将近一秒，不知道做KV的缓存来加速是否可行