set dtype to float as arange returns int32 by default.