1、__ init __()是什么?

从形式上看,__ init __()是一个函数。事实上,它是一个特殊的函数——构造函数,学过C或Java的应该都知道构造函数,没学过也没关系,构造函数相当于是对类中成员的一个初始化,且写法固定

2、如何写 __ init __()?

__ init __()Python中类里面的构造函数,因此,写法固定。以一个例子来说明__ init __()的写法:

class EncoderDecoder(nn.Module):
    """
    A standard Encoder-Decoder architecture. Base for this and many 
    other models.
    """
    def __init__(self, encoder, decoder, src_embed, tgt_embed, generator):
        super(EncoderDecoder, self).__init__()
        self.encoder = encoder
        self.decoder = decoder
        self.src_embed = src_embed
        self.tgt_embed = tgt_embed
        self.generator = generator
        
    def forward(self, src, tgt, src_mask, tgt_mask):
        "Take in and process masked src and target sequences."
        return self.decode(self.encode(src, src_mask), src_mask,
                            tgt, tgt_mask)
    
    def encode(self, src, src_mask):
        return self.encoder(self.src_embed(src), src_mask)
    
    def decode(self, memory, src_mask, tgt, tgt_mask):
        return self.decoder(self.tgt_embed(tgt), memory, src_mask, tgt_mask)

这段代码创建了一个类EncoderDecoder,该类用来建模一个标准的编码器-解码器架构,这里仅关注__init__(self, encoder, decoder, src_embed, tgt_embed, generator)这个函数。
可以看到参数列表中,有self, encoder, decoder, src_embed, tgt_embed, generator共6个参数,其中,self表示类EncoderDecoder实例化成对象以后这个对象本身,其他5个是用户自定义从外部传入的参数,与编码器-解码器架构中的相关属性一一对应。
super(EncoderDecoder, self).__init__()表示继承父类nn.Module中的__ init __()方法。
self.encoder = encoder表示将传入的encoder这个参数赋值给对象的属性变量,即self.encoder。剩下的几行代码以此类推。

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐