Rui An
convert norm function output to original dtype
7711170