* Fix scatter_op python API and remove inconsistency between implementation and doc

* API spec change

* Change as review comment