| import math | |
| import torch | |
| def rounded_linspace(low, high, steps, div): | |
| ret = torch.linspace(low, high, steps) | |
| ret = (ret.int() + div - 1) // div * div | |
| ret = torch.unique(ret) | |
| return list(map(int, ret)) | |
| def powspace(start, stop, pow, step): | |
| start = math.log(start, pow) | |
| stop = math.log(stop, pow) | |
| steps = int((stop - start + 1) // step) | |
| ret = torch.pow(pow, torch.linspace(start, stop, steps)) | |
| ret = torch.unique(ret) | |
| return list(map(int, ret)) |