torch中三种sigmoid使用方法

关于下面三种sigmoid的使用方法

  • torch.sigmoid()
  • torch.nn.functional.sigmoid()
  • torch.Sigmoid
相同点
  • 都是将值映射到0-1之间,没用区别
不同点
  1. 所属类型不同
    方法:torch.sigmoid() 和torch.nn.funtional.sigmoid()
    :torch.Sigmoid
  2. 使用方法不同
    两个方法函数直接传输向量就行,而类需要先定义一个类,然后再通过调用类本身的__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()
torch中三种sigmoid使用方法_第1张图片

你可能感兴趣的:(pytorch)