关于下面三种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()
在这里插入图片描述

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐