nn.Sigmoid torch

from torch import nn
import torch


test = torch.tensor([1, 5., 4, 8, 9])
s = nn.Sigmoid()    
print(s(test))
    
test = torch.tensor([[1, 5., 4, 8, 9],[1, 6., 4, 2, 7]])
s = nn.Sigmoid()
print(s(test))


結果:
tensor([0.7311, 0.9933, 0.9820, 0.9997, 0.9999])
tensor([[0.7311, 0.9933, 0.9820, 0.9997, 0.9999],
        [0.7311, 0.9975, 0.9820, 0.8808, 0.9991]])

公式爲:

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章