Dim<0> is for scalar (rank-0 tensor). Adding Dim<0> can simplify
a lot of code.