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