PyTorch torch.addcdiv 関数


Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

torch.addcdivは PyTorch においてinput + value * (tensor1 / tensor2)要素ごとの演算を実行する関数です。

関数定義

torch.addcdiv(input, tensor1, tensor2, value=1, out=None)

引数の説明

  • input: 入力テンソル
  • tensor1: 分子テンソル
  • tensor2: 分母テンソル(ゼロにできない)
  • value: 乗数、デフォルトは 1
  • out: 出力テンソル(オプション)

使用例

実例

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)

出力結果は次のとおりです:

tensor([3., 6., 9.])

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張