Fix api doc example bug and polish square doc (#20491)

test=develop
test=document_fix

* Refine topk stack squeeze api en doc

test=develop
test=document_fix

* refine api.spec and returns

test=develop
test=document_fix
Signed-off-by: Nzhaoyuchen <zhaoyuchen01@baidu.com>