PyTorch torch.addcdiv 関数
torch.addcdivは PyTorch においてinput + value * (tensor1 / tensor2)要素ごとの演算を実行する関数です。
関数定義
torch.addcdiv(input, tensor1, tensor2, value=1, out=None)
引数の説明
input: 入力テンソルtensor1: 分子テンソルtensor2: 分母テンソル(ゼロにできない)value: 乗数、デフォルトは 1out: 出力テンソル(オプション)
使用例
実例
import torch
# テンソルを作成
input = torch.tensor([1.0, 2.0, 3.0])
tensor1 = torch.tensor([4.0, 8.0, 12.0])
tensor2 = torch.tensor([2.0, 2.0, 2.0])
# input + value * (tensor1 / tensor2) を実行
result = torch.addcdiv(input, tensor1, tensor2, value=1)
print(result)
# テンソルを作成
input = torch.tensor([1.0, 2.0, 3.0])
tensor1 = torch.tensor([4.0, 8.0, 12.0])
tensor2 = torch.tensor([2.0, 2.0, 2.0])
# input + value * (tensor1 / tensor2) を実行
result = torch.addcdiv(input, tensor1, tensor2, value=1)
print(result)
出力結果は次のとおりです:
tensor([3., 6., 9.])
その他の拡張
Pytorch torch リファレンスマニュアル