当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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 张量。

startend 的形状必须是可广播的。如果 weight 是张量,则 weightstartend 的形状必须是可广播的。

例子:

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

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.lerp。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。