比较与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]