Функция logspace
Функция logspace создает одномерный тензор, содержащий
равномерно распределенные значения на логарифмической шкале
между start и end. Первым параметром функция
принимает начальное значение степени, вторым - конечное
значение степени, третьим - количество элементов.
Основание логарифма по умолчанию равно 10, но его
можно изменить параметром base.
Синтаксис
torch.logspace(start, end, [steps], [base], [dtype])
Пример
Давайте создадим тензор из 5 чисел на логарифмической шкале
от 10⁰ до 10⁴:
import torch
t = torch.logspace(0, 4, steps=5)
print(t)
Результат выполнения кода:
tensor([1.0000e+00, 1.0000e+01, 1.0000e+02, 1.0000e+03, 1.0000e+04])
Пример
Теперь создадим тензор с основанием 2 от 2⁰ до 2³:
import torch
t = torch.logspace(0, 3, steps=4, base=2)
print(t)
Результат выполнения кода:
tensor([1., 2., 4., 8.])
Пример
Создадим тензор с типом float на логарифмической шкале
от 10⁻¹ до 10²:
import torch
t = torch.logspace(-1, 2, steps=4, dtype=torch.float)
print(t)
Результат выполнения кода:
tensor([0.1000, 1.0000, 10.0000, 100.0000])