Keras 3.10新增remat实现内存高效训练

文档：
https://keras.io/api/rematerialization/remat/
...

你可以在自定义层中使用keras.remat，或者使用keras.remat_scope(mode)选择性地应用于任何现有层/模型。