比较与torch.expm1的功能差异

torch.expm1

torch.expm1(input, *, out=None) -> Tensor

更多内容详见torch.expm1

mindspore.ops.expm1

mindspore.ops.expm1(x) -> Tensor

更多内容详见mindspore.ops.expm1

差异对比

PyTorch:逐元素计算输入张量的指数减1的值。

MindSpore:与PyTorch实现同样的功能,仅参数名不同。

分类

子类

PyTorch

MindSpore

差异

参数

参数1

input

x

功能一致,参数名不同

参数2

out

-

不涉及

代码示例1

两API实现功能相同,用法相同

# PyTorch
import torch
from torch import tensor
import numpy as np

inputx_ = np.array([0.0, 1.0, 2.0, 4.0])
inputx = tensor(inputx_, dtype=torch.float32)
output = torch.expm1(inputx)
output_m = output.detach().numpy()
print(output_m)
#[ 0.         1.7182817  6.389056  53.59815  ]

# MindSpore
import mindspore
from mindspore import Tensor
import mindspore.ops as ops
import numpy as np

x_ = np.array([0.0, 1.0, 2.0, 4.0])
x = Tensor(x_, mindspore.float32)
output = ops.expm1(x)
print(output)
#[0.        1.7182819  6.389056  53.598152]