torch.lerp¶
- torch.lerp(input, end, weight, *, out=None)¶
根据标量或张量
weight
对两个张量start
(由input
给出)和end
进行线性插值,并返回结果张量out
。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])