torch.lerp

torch.lerp(input, end, weight, *, out=None)

Выполняет линейную интерполяцию двух тензоров start (заданных input) и end на основе скалярного или тензорного значения weight и возвращает результирующий out тензор.

$$\text{out}_i = \text{start}_i + \text{weight}_i \times (\text{end}_i - \text{start}_i)$$

Формы start и end должны быть совместимы для трансляции. Если weight является тензором, то формы weight, start, и end должны быть совместимы для трансляции.

Параметры:

• input (Tensor) – тензор с начальными точками
• end (Tensor) – тензор с конечными точками
• weight (float или тензор) – вес для формулы интерполяции

Ключевые аргументы:

out (Tensor, необязательно) – выходной тензор.

Пример:

>>> start = torch.arange(1., 5.)
>>> end = torch.empty(4).fill_(10)
>>> start
tensor([ 1.,  2.,  3.,  4.])
>>> end
tensor([ 10.,  10.,  10.,  10.])
>>> torch.lerp(start, end, 0.5)
tensor([ 5.5000,  6.0000,  6.5000,  7.0000])
>>> torch.lerp(start, end, torch.full_like(start, 0.5))
tensor([ 5.5000,  6.0000,  6.5000,  7.0000])