* update dist tensor

* add unitest

* update unitest

* refactor dist tensor

* update dist tensor and unitest