* update gather_tree doc, test=document_fix

* update sample code, test=document_fix

* remove tensor type, test=document_fix