PyTorch torch.trapz 関数
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)
# 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 リファレンスマニュアル