torch中三种sigmoid使用方法
关于下面三种sigmoid的使用方法torch.sigmoid()torch.nn.functional.sigmoid()torch.Sigmoid相同点都是将值映射到0-1之间,没用区别不同点所属类型不同方法:torch.sigmoid() 和torch.nn.funtional.sigmoid()类:torch.Sigmoid使用方法不同两个方法函数直接传输向量就行,而类需要先定义一个类,然
·
关于下面三种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()
更多推荐



所有评论(0)