PyTorch torch.trapz 関数


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

torch.trapzは PyTorch で台形積分を計算するための関数であり、torch.trapezoid関数のエイリアスで、両者の機能は同じです。

関数定義

torch.trapz(y, x, dx)

パラメータ説明

  • y: 積分する関数値
  • x: 積分変数の値(オプション)
  • dx: サンプリング間隔(x が指定されない場合に使用)

使用例

実例

import torch

# dx を使用した台形積分
y = torch.tensor([1.0, 2.0, 3.0, 4.0])

# dx=1.0 は均一なサンプリング間隔を表す
result = torch.trapz(y, dx=1.0)

print("台形積分結果 (trapz):", result)

# trapezoid と同等
result2 = torch.trapezoid(y, dx=1.0)
print("台形積分結果 (trapezoid):", result2)

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

梯形积分结果 (trapz): tensor(7.5000)
梯形积分结果 (trapezoid): tensor(7.5000)

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

その他の拡張