pytorch torch.sigmoid、torch.nn.Sigmoid和torch.nn.functional.sigmoid的区别

torch.sigmoid:函数

torch.sigmoid(input_tensor,out=None)->Tensor

torch.nn.Sigmoid:class

应该看作网络的一层,而不是简单的函数使用。
模型的初始化方法中使用,需要在_init__中定义,然后在使用

class Model(torch.nn.Module):
    def __init__(self):
        super(Model, self

你可能感兴趣的:(pytorch torch.sigmoid、torch.nn.Sigmoid和torch.nn.functional.sigmoid的区别)