you can put the dtype in np.arange

```
s = math.sqrt(2. / (n_times + 1.))
tmp = np.arange(1, n_times + 1, dtype=np.float64)
out_x = tmp[None, :]  * tmp[:, None]
```