Функция torch.expm1 в PyTorch

Функция torch.expm1 в PyTorch


Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.expm1Это функция в PyTorch, используемая для поэлементного вычисления exp(x) - 1.

Определение функции

torch.expm1(input, *, out=None)

Пример использования

Пример

import torch

# пример exp(x) - 1
x = torch.tensor([0, 1, 2, 3], dtype=torch.float32)
result = torch.expm1(x)
print(result)  # tensor([0.0000, 1.7183, 6.3891, 19.0855])

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения