Align with dpt.where
which support scalar as input
#1932
Merged
dpt.where
which support scalar as input
#1932