fix model doc str: add shape hint to doc string. by wanghan-iapcm · Pull Request #3225 · deepmodeling/deepmd-kit