关于下面三种sigmoid的使用方法
- torch.sigmoid()
- torch.nn.functional.sigmoid()
- torch.Sigmoid
相同点
- 都是将值映射到0-1之间,没用区别
不同点
- 所属类型不同
方法:torch.sigmoid() 和torch.nn.funtional.sigmoid()
类:torch.Sigmoid - 使用方法不同
两个方法函数直接传输向量就行,而类需要先定义一个类,然后再通过调用类本身的__call__函数去使用。
a = torch.tensor([1,2,3])
print(torch.sigmoid(a))
print(F.sigmoid(a))
b = nn.Sigmoid()
print(b(a))
同时,系统推荐使用nn.sigmoid()而不是nn.functional.sigmoid()