20 lines
488 B
Python
20 lines
488 B
Python
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))
|