PyTorch torch.addcmul 関数
Pytorch torch リファレンスマニュアルtorch.addcmulは PyTorch で実行するために使用されるinput + value * (tensor1 * tensor2)要素ごとの演算の関数です。
関数定義
torch.addcmul(input, tensor1, tensor2, value=1, out=None)
パラメータ説明
input: 入力テンソルtensor1: 最初の乗数テンソルtensor2: 2番目の乗数テンソルvalue: 乗数、デフォルトは1out: 出力テンソル（オプション）
使用例
例
import torch
# テンソルを作成
input = torch.tensor([1.0, 2.0, 3.0])
tensor1 = torch.tensor([2.0, 3.0, 4.0])
tensor2 = torch.tensor([3.0, 4.0, 5.0])
# input + value * (tensor1 * tensor2) を実行
result = torch.addcmul(input, tensor1, tensor2, value=1)
print(result)
# テンソルを作成
input = torch.tensor([1.0, 2.0, 3.0])
tensor1 = torch.tensor([2.0, 3.0, 4.0])
tensor2 = torch.tensor([3.0, 4.0, 5.0])
# input + value * (tensor1 * tensor2) を実行
result = torch.addcmul(input, tensor1, tensor2, value=1)
print(result)
出力結果は：
tensor([ 7., 14., 23.])
その他の拡張