* add n-d input support for trt scale converter

* add flatten for ut

* fix dims