[torchlib] Fix aten::arange to support dynamic shapes (#1913)

[torchlib] Fix aten::arange to support dynamic shapes (#1913) #855