* support non-tensor type

* fix unittest failed

* add unittest with prune

* rm unused code

* coverage

* fix two or