import torch
import torch.nn as nn
a = torch.arange(4).reshape(2,2).float()
print('a',a)
b =nn.Sigmoid()
print(b(a))
这个nn.sigmoid()完成的是对位置内每一个都按公式计算
a tensor([[0., 1.],
[2., 3.]])
tensor([[0.5000, 0.7311],
[0.8808, 0.9526]])
import torch
import torch.nn as nn
a = torch.arange(4).reshape(2,2).float()
print('a',a)
b =nn.Sigmoid()
print(b(a))
这个nn.sigmoid()完成的是对位置内每一个都按公式计算
a tensor([[0., 1.],
[2., 3.]])
tensor([[0.5000, 0.7311],
[0.8808, 0.9526]])