比较与torch.nn.Tanh的功能差异

torch.nn.Tanh

class torch.nn.Tanh()(input) -> Tensor

更多内容详见torch.nn.Tanh。

mindspore.nn.Tanh

class mindspore.nn.Tanh()(x) -> Tensor

更多内容详见mindspore.nn.Tanh。

差异对比

PyTorch：计算双曲正切函数tanh。

MindSpore：MindSpore此API实现功能与PyTorch一致。

分类	子类	PyTorch	MindSpore	差异
输入	单输入	input	x	功能一致，参数名不同

代码示例

计算输入x的tanh函数，MindSpore此API功能与PyTorch一致。

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

m = nn.Tanh()
x_ = np.array([0.7713, 0.0208, 0.6336], dtype=np.float32)
x = tensor(x_)
output = m(x)
print(output.numpy())
# [0.64768475 0.020797   0.56052613]

# MindSpore
import numpy as np
import mindspore
from mindspore import Tensor, nn

m = nn.Tanh()
x_ = np.array([0.7713, 0.0208, 0.6336], dtype=np.float32)
x = Tensor(x_, mindspore.float32)
output = m(x)
print(output)
# [0.64768475 0.020797   0.56052613]