Python PyTorch lerp用法及代碼示例

本文簡要介紹python語言中 torch.lerp 的用法。

用法:

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

參數：

• input(Tensor) -具有起點的張量

• end(Tensor) -具有端點的張量

• weight(float或者張量) -插值公式的權重

關鍵字參數：

out(Tensor,可選的) -輸出張量。

根據標量或張量 weight 對兩個張量 start(由 input 給出)和 end 進行線性插值，並返回結果 out 張量。

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

start 和 end 的形狀必須是可廣播的。如果 weight 是張量，則 weight 、 start 和 end 的形狀必須是可廣播的。

例子：

>>> 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])