[Feature Request] Support `stat_file` in TF

[njzjz]

Summary

Support training/stat_file in TF.

Detailed Description

It's now a PT-only option.

Further Information, Files, and Links

https://docs.deepmodeling.com/projects/deepmd/en/latest/train/train-input.html#argument:training/stat_file