PyTorch torch.igamma 函数
torch.igamma 是 PyTorch 中用于计算逐元素不完全 gamma 函数的函数。
函数定义
torch.igamma(input, other, out=None)
使用示例
实例
import torch
# 计算不完全 gamma 函数
x = torch.tensor([1.0, 2.0, 3.0])
a = torch.tensor([1.0, 1.0, 1.0])
result = torch.igamma(a, x)
print(result)
# 计算不完全 gamma 函数
x = torch.tensor([1.0, 2.0, 3.0])
a = torch.tensor([1.0, 1.0, 1.0])
result = torch.igamma(a, x)
print(result)

Pytorch torch 参考手册