PyTorch torch.lerp 関数
torch.lerpPyTorch で2つのテンソル間の線形補間を計算するために使用される関数です。
関数定義
torch.lerp(input, end, weight, out=None)
使用例
例
import torch
# 2つのテンソルを作成
start = torch.tensor([0.0, 0.0])
end = torch.tensor([10.0, 20.0])
# weight=0.5 で線形補間を実行
result = torch.lerp(start, end, 0.5)
print(「補間結果:」, result)
# 2つのテンソルを作成
start = torch.tensor([0.0, 0.0])
end = torch.tensor([10.0, 20.0])
# weight=0.5 で線形補間を実行
result = torch.lerp(start, end, 0.5)
print(「補間結果:」, result)
出力結果は:
插值结果: tensor([ 5., 10.])
その他の拡張
Pytorch torch リファレンスマニュアル